Make map_coordinates
differentiable for JAX 0.4.34
#7375
Job | Run time |
---|---|
1m 25s | |
1m 25s |
map_coordinates
differentiable for JAX 0.4.34
#7375
Job | Run time |
---|---|
1m 25s | |
1m 25s |