From c2d752ec4b2d62a4727235eae55f5d63e2160cca Mon Sep 17 00:00:00 2001 From: ivyzheng Date: Wed, 7 Feb 2024 13:29:57 -0800 Subject: [PATCH] release 0.8.1 --- CHANGELOG.md | 12 ++++++++++++ README.md | 2 +- 2 files changed, 13 insertions(+), 1 deletion(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index 03715e8513..894155d337 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -24,6 +24,18 @@ vNext - - +0.8.1 +----- +- Added default collection in `make_rng`. +- Added `InstanceNorm` and renamed `channel_axes` to `feature_axes`. +- Added norm equivalence tests. +- Added `Module.module_paths` and doc. +- make `Sequential.__call__` compact. +- Added `nn.compact_name_scope` v3. +- Add explicit control over frozen/slots setting in `flax.struct.dataclass`. +- Replacing `jax.tree_util.tree_map` with mapping over leafs. +- Fixed docs and docstrings. + 0.8.0 ----- - Added [NNX](https://github.com/google/flax/tree/main/flax/experimental/nnx#nnx), a neural network library for JAX that provides a simple yet powerful module system that adheres to standard Python semantics. Its aim is to combine the robustness of Linen with a simplified, Pythonic API akin to that of PyTorch. diff --git a/README.md b/README.md index 4630ee9de6..0adf8a9cca 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.8.0}, + version = {0.8.1}, year = {2023}, } ```