Flax will default to returning regular dicts instead of FrozenDicts in the future. Some internal-only tests fail because they expect frozen params when creating TrainState.#3104
Closed
copybara-service[bot] wants to merge 0 commit intomainfrom test_533567519
Commits
No commits history
There isn't any commit history to show here.