Make map_coordinates
differentiable for JAX 0.4.34
#664
Job | Run time |
---|---|
15m 58s | |
24m 59s | |
30m 7s | |
1h 11m 4s |
map_coordinates
differentiable for JAX 0.4.34
#664
Job | Run time |
---|---|
15m 58s | |
24m 59s | |
30m 7s | |
1h 11m 4s |