091f91d061
Diff: https://github.com/deepmind/optax/compare/refs/tags/v0.2.2...v0.2.3 Changelog: https://github.com/deepmind/optax/releases/tag/v0.2.3