Make map_coordinates
differentiable for JAX 0.4.34
#665
Job | Run time |
---|---|
16m 34s | |
10m 45s | |
12m 26s | |
14m 50s | |
14m 35s | |
18m 4s | |
16m 39s | |
21m 1s | |
2h 4m 54s |
map_coordinates
differentiable for JAX 0.4.34
#665
Job | Run time |
---|---|
16m 34s | |
10m 45s | |
12m 26s | |
14m 50s | |
14m 35s | |
18m 4s | |
16m 39s | |
21m 1s | |
2h 4m 54s |