Make map_coordinates
differentiable for JAX 0.4.34
#351
Loading
map_coordinates
differentiable for JAX 0.4.34
#351