Migrating Flax from returning FrozenDicts to returning regular dicts. More details can be found in this [announcement](https://github.com/google/flax/discussions/3191) #9786
Job | Run time |
---|---|
16s | |
5s | |
4s | |
45s | |
33s | |
34s | |
1s | |
1s | |
1s | |
1s | |
1s | |
1s | |
1s | |
1s | |
1s | |
2m 26s |