Changes: https://github.com/google/flax/releases https://github.com/google/flax/blob/master/CHANGELOG.md
