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