Migrating Flax from returning FrozenDicts to returning regular dicts. More details can be found in this [announcement](https://github.com/google/flax/discussions/3191) #9771
Job | Run time |
---|---|
5s | |
19s | |
3s | |
36s | |
39s | |
42s | |
1s | |
1s | |
1s | |
7s | |
7s | |
7s | |
8s | |
8s | |
6s | |
3m 10s |