Closed esmagden closed 1 year ago
Jax is required for map_coordinates, but is trax also in fact a requirement?
It feels like it shouldn't be, and we should remove it if that's the case
Trax is a requirement because trax.fastmath.grad is used in effective_index module.
trax.fastmath.grad
effective_index
trax was removed
trax
Jax is required for map_coordinates, but is trax also in fact a requirement?
It feels like it shouldn't be, and we should remove it if that's the case