Let Flax-Orbax to not port the shape of target
arrays when they por…
#9847
Job | Run time |
---|---|
4s | |
5s | |
23s | |
49s | |
37s | |
43s | |
6m 49s | |
7m 33s | |
8m 56s | |
4m 33s | |
6m 40s | |
7m 31s | |
4m 42s | |
8m 59s | |
9m 22s | |
1h 7m 46s |
target
arrays when they por…
#9847
Job | Run time |
---|---|
4s | |
5s | |
23s | |
49s | |
37s | |
43s | |
6m 49s | |
7m 33s | |
8m 56s | |
4m 33s | |
6m 40s | |
7m 31s | |
4m 42s | |
8m 59s | |
9m 22s | |
1h 7m 46s |