Make map_coordinates
differentiable for JAX 0.4.34
#674
Job | Run time |
---|---|
15m 20s | |
24m 44s | |
22m 15s | |
16m 21s | |
30m 44s | |
16m 16s | |
2h 5m 40s |
map_coordinates
differentiable for JAX 0.4.34
#674
Job | Run time |
---|---|
15m 20s | |
24m 44s | |
22m 15s | |
16m 21s | |
30m 44s | |
16m 16s | |
2h 5m 40s |