Make map_coordinates
differentiable for JAX 0.4.34
#333
Job | Run time |
---|---|
30s | |
30s | |
32s | |
29s | |
1m 19s | |
1m 19s | |
1m 22s | |
1m 18s | |
1m 16s | |
1m 19s | |
1m 13s | |
1m 13s | |
1m 12s | |
1m 21s | |
1m 11s | |
1m 15s | |
1m 14s | |
1m 15s | |
1m 16s | |
1m 12s | |
1m 24s | |
1m 16s | |
1m 16s | |
1m 17s | |
32m 18s | |
31m 15s | |
31m 55s | |
31m 10s | |
29m 34s | |
27m 55s | |
15m 56s | |
13m 52s | |
30m 42s | |
26m 59s | |
33m 36s | |
32m 20s | |
30m 36s | |
28m 16s | |
29m 56s | |
29m 18s | |
28m 3s | |
26m 51s | |
6m 25s | |
27m 35s | |
9h 32m 1s |