I have been playing around with converting diffmpm from the difftaichi package into a jax version, and while the forward pass has been working wonderfully, the backward pass has been using way too ...

init_cond.bit_depth_image = double(h5read(master_fp,'/entry/instrument/detector/bit_depth_image')); init_cond.bad_point_value = power(2,init_cond.bit_depth_image)-1 ...