Merge pull request #3269 from chiamp:release

PiperOrigin-RevId: 555005843
google · Aug 9, 2023 · d826006 · d826006
2 parents fe54d39 + 5eeb506
commit d826006
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git 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
@@ -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},
 }
 ```