jax_implementation# Functions full_recon(t, geometry, dl, nx, ny, nz, dx, ...) Do delay and sum for all detectors. recon_partial(t, geometry, dl, nx, ny, nz, ...) Do delay and sum for a single detector.