Make map_coordinates differentiable for JAX 0.4.34
#1293
Merged
map_coordinates differentiable for JAX 0.4.34
#1293