Let Flax-Orbax to not port the shape of target
arrays when they por…
#9837
Job | Run time |
---|---|
8s | |
2s | |
46s | |
33s | |
39s | |
30s | |
3m 58s | |
3m 32s | |
3m 35s | |
3m 31s | |
4m 6s | |
4m 2s | |
3m 45s | |
3m 59s | |
3m 55s | |
37m 1s |