Diff: https://github.com/google/flax/compare/refs/tags/v0.8.3...v0.8.4 Changelog: https://github.com/google/flax/releases/tag/v0.8.4