Make map_coordinates
differentiable for JAX 0.4.34
#7336
Job | Run time |
---|---|
1m 28s | |
1m 28s |
map_coordinates
differentiable for JAX 0.4.34
#7336
Job | Run time |
---|---|
1m 28s | |
1m 28s |