Make map_coordinates
differentiable for JAX 0.4.34
#530
Job | Run time |
---|---|
7m 4s | |
7m 2s | |
6m 44s | |
7m 5s | |
7m 2s | |
7m 9s | |
7m 0s | |
5m 10s | |
54m 16s |
map_coordinates
differentiable for JAX 0.4.34
#530
Job | Run time |
---|---|
7m 4s | |
7m 2s | |
6m 44s | |
7m 5s | |
7m 2s | |
7m 9s | |
7m 0s | |
5m 10s | |
54m 16s |