diff --git a/CHANGELOG.md b/CHANGELOG.md index 09fbf3a074..3e8ffc800e 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -23,6 +23,17 @@ vNext - - +0.7.2 +----- +New features: +- make `flax.core.copy` `add_or_replace` optional +- Add `use_fast_variance` option to `GroupNorm` and `BatchNorm` to allow disabling it. + +Bug fixes: +- Use `field_specifiers` instead of `field_descriptors` in `@dataclass_transform`. +- Fix `nn.Module` typing. +- [JAX] Replace uses of `jax.experimental.pjit.with_sharding_constraint` with `jax.lax.with_sharding_constraint`. + 0.7.1 ----- Breaking changes: diff --git a/README.md b/README.md index ac7b9433ba..b70e35c875 100644 --- a/README.md +++ b/README.md @@ -197,7 +197,7 @@ To cite this repository: author = {Jonathan Heek and Anselm Levskaya and Avital Oliver and Marvin Ritter and Bertrand Rondepierre and Andreas Steiner and Marc van {Z}ee}, title = {{F}lax: A neural network library and ecosystem for {JAX}}, url = {http://github.com/google/flax}, - version = {0.7.1}, + version = {0.7.2}, year = {2023}, } ```