Make map_coordinates
differentiable for JAX 0.4.34
#527
Job | Run time |
---|---|
10m 47s | |
10m 27s | |
10m 49s | |
10m 46s | |
10m 47s | |
10m 45s | |
10m 51s | |
10m 47s | |
1h 25m 59s |
map_coordinates
differentiable for JAX 0.4.34
#527
Job | Run time |
---|---|
10m 47s | |
10m 27s | |
10m 49s | |
10m 46s | |
10m 47s | |
10m 45s | |
10m 51s | |
10m 47s | |
1h 25m 59s |