Make map_coordinates
differentiable for JAX 0.4.34
#8903
Job | Run time |
---|---|
32m 32s | |
40m 19s | |
1h 12m 51s |
map_coordinates
differentiable for JAX 0.4.34
#8903
Job | Run time |
---|---|
32m 32s | |
40m 19s | |
1h 12m 51s |