Make map_coordinates
differentiable for JAX 0.4.34
#698
Job | Run time |
---|---|
17m 47s | |
11m 29s | |
12m 20s | |
14m 32s | |
15m 26s | |
19m 0s | |
15m 39s | |
20m 21s | |
2h 6m 34s |
map_coordinates
differentiable for JAX 0.4.34
#698
Job | Run time |
---|---|
17m 47s | |
11m 29s | |
12m 20s | |
14m 32s | |
15m 26s | |
19m 0s | |
15m 39s | |
20m 21s | |
2h 6m 34s |