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
test_533567519% was force-pushed and no longer has any new commits.
Pushing new commits will allow the pull request to be re-opened.