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