Make map_coordinates
differentiable for JAX 0.4.34
#698
Job | Run time |
---|---|
15m 46s | |
25m 10s | |
22m 3s | |
16m 25s | |
31m 15s | |
16m 18s | |
2h 6m 57s |
map_coordinates
differentiable for JAX 0.4.34
#698
Job | Run time |
---|---|
15m 46s | |
25m 10s | |
22m 3s | |
16m 25s | |
31m 15s | |
16m 18s | |
2h 6m 57s |