Make map_coordinates
differentiable for JAX 0.4.34
#662
Job | Run time |
---|---|
16m 22s | |
25m 6s | |
31m 52s | |
1h 13m 20s |
map_coordinates
differentiable for JAX 0.4.34
#662
Job | Run time |
---|---|
16m 22s | |
25m 6s | |
31m 52s | |
1h 13m 20s |