Make map_coordinates
differentiable for JAX 0.4.34
#656
Job | Run time |
---|---|
15m 56s | |
18m 33s | |
18m 16s | |
52m 45s |
map_coordinates
differentiable for JAX 0.4.34
#656
Job | Run time |
---|---|
15m 56s | |
18m 33s | |
18m 16s | |
52m 45s |