Make map_coordinates
differentiable for JAX 0.4.34
#664
Job | Run time |
---|---|
16m 23s | |
11m 30s | |
12m 42s | |
14m 30s | |
14m 20s | |
17m 57s | |
17m 2s | |
20m 1s | |
2h 4m 25s |
map_coordinates
differentiable for JAX 0.4.34
#664
Job | Run time |
---|---|
16m 23s | |
11m 30s | |
12m 42s | |
14m 30s | |
14m 20s | |
17m 57s | |
17m 2s | |
20m 1s | |
2h 4m 25s |