From 077a165210073fe78f8df9cf1b6ced1b5fc7ff24 Mon Sep 17 00:00:00 2001 From: Marc van Zee Date: Thu, 5 May 2022 12:40:22 +0200 Subject: [PATCH 1/2] Bumps Flax to v0.4.2 and adds CHANGELOG notes --- CHANGELOG.md | 39 +++++++++++++++++++++++++++++++++++++-- README.md | 2 +- flax/version.py | 2 +- 3 files changed, 39 insertions(+), 4 deletions(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index dfda17bc4..92a076d00 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -8,8 +8,6 @@ vNext - - - -- Added lifted conditional `nn.cond` -- Handle rate==1.0 edgecase in Dropout. - - - @@ -17,6 +15,43 @@ vNext - - - +- +- +- +- +- +- +- +- +- +- + +0.4.2 +----- + +New features: +* Canonicalize conv padding by @jheek in https://github.com/google/flax/pull/2009 +* Update ScopeParamNotFoundError message. by @melissatan in https://github.com/google/flax/pull/2013 +* Set field on dataclass transform decorator by @NeilGirdhar in https://github.com/google/flax/pull/1927 +* Don't recommend mixing setup and compact in docs. by @levskaya in https://github.com/google/flax/pull/2018 +* Clarifies `optim.Adam(weight_decay)` parameter. by @copybara-service in https://github.com/google/flax/pull/2016 +* Update linear regression example in Jax intro and Flax intro. by @melissatan in https://github.com/google/flax/pull/2015 +* Lifted cond by @jheek in https://github.com/google/flax/pull/2020 +* Use tree_map instead of deprecated tree_multimap by @jheek in https://github.com/google/flax/pull/2024 +* Remove tree_multimap from docs, examples, and tests by @jheek in https://github.com/google/flax/pull/2026 +* Fix bug where the linen Module state is reused. by @jheek in https://github.com/google/flax/pull/2025 +* Add __getattribute__ with lazy setup trigger. by @levskaya in https://github.com/google/flax/pull/2028 +* Better error messages for loading checkpoints. by @copybara-service in https://github.com/google/flax/pull/2035 +* Add filterwarning for jax.tree_multimap by @marcvanzee in https://github.com/google/flax/pull/2038 +* Adds Flax logo to README by @marcvanzee in https://github.com/google/flax/pull/2036 +* Module lifecycle note by @jheek in https://github.com/google/flax/pull/1964 +* Fix linter errors in core/scope.py and core/tracers.py. by @copybara-service in https://github.com/google/flax/pull/2004 +* Handle edge-case of rate==1.0 in Dropout layer. by @levskaya in https://github.com/google/flax/pull/2055 +* Bug fixes and generalizations of nn.partitioning api. by @copybara-service in https://github.com/google/flax/pull/2062 +* Add support for JAX dynamic stack-based named_call. by @copybara-service in https://github.com/google/flax/pull/2063 +* Updates pooling docstrings by @marcvanzee in https://github.com/google/flax/pull/2064 +* Makes annotated_mnist use Optax's xent loss. by @andsteing in https://github.com/google/flax/pull/2071 + 0.4.1 ----- diff --git a/README.md b/README.md index bfc923f60..649c77fdd 100644 --- a/README.md +++ b/README.md @@ -190,7 +190,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.4.1}, + version = {0.4.2}, year = {2020}, } ``` diff --git a/flax/version.py b/flax/version.py index 6c5e36c27..af09a08bb 100644 --- a/flax/version.py +++ b/flax/version.py @@ -13,5 +13,5 @@ # limitations under the License. """Current Flax version at head on Github.""" -__version__ = "0.4.1" +__version__ = "0.4.2" From d56f2e282245e09931e16e701d17ed01200d5767 Mon Sep 17 00:00:00 2001 From: Marc van Zee Date: Thu, 5 May 2022 13:13:18 +0200 Subject: [PATCH 2/2] Updates CHANGELOG.md --- CHANGELOG.md | 34 +++++++++++++--------------------- 1 file changed, 13 insertions(+), 21 deletions(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index 92a076d00..4ec34599b 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -30,27 +30,19 @@ vNext ----- New features: -* Canonicalize conv padding by @jheek in https://github.com/google/flax/pull/2009 -* Update ScopeParamNotFoundError message. by @melissatan in https://github.com/google/flax/pull/2013 -* Set field on dataclass transform decorator by @NeilGirdhar in https://github.com/google/flax/pull/1927 -* Don't recommend mixing setup and compact in docs. by @levskaya in https://github.com/google/flax/pull/2018 -* Clarifies `optim.Adam(weight_decay)` parameter. by @copybara-service in https://github.com/google/flax/pull/2016 -* Update linear regression example in Jax intro and Flax intro. by @melissatan in https://github.com/google/flax/pull/2015 -* Lifted cond by @jheek in https://github.com/google/flax/pull/2020 -* Use tree_map instead of deprecated tree_multimap by @jheek in https://github.com/google/flax/pull/2024 -* Remove tree_multimap from docs, examples, and tests by @jheek in https://github.com/google/flax/pull/2026 -* Fix bug where the linen Module state is reused. by @jheek in https://github.com/google/flax/pull/2025 -* Add __getattribute__ with lazy setup trigger. by @levskaya in https://github.com/google/flax/pull/2028 -* Better error messages for loading checkpoints. by @copybara-service in https://github.com/google/flax/pull/2035 -* Add filterwarning for jax.tree_multimap by @marcvanzee in https://github.com/google/flax/pull/2038 -* Adds Flax logo to README by @marcvanzee in https://github.com/google/flax/pull/2036 -* Module lifecycle note by @jheek in https://github.com/google/flax/pull/1964 -* Fix linter errors in core/scope.py and core/tracers.py. by @copybara-service in https://github.com/google/flax/pull/2004 -* Handle edge-case of rate==1.0 in Dropout layer. by @levskaya in https://github.com/google/flax/pull/2055 -* Bug fixes and generalizations of nn.partitioning api. by @copybara-service in https://github.com/google/flax/pull/2062 -* Add support for JAX dynamic stack-based named_call. by @copybara-service in https://github.com/google/flax/pull/2063 -* Updates pooling docstrings by @marcvanzee in https://github.com/google/flax/pull/2064 -* Makes annotated_mnist use Optax's xent loss. by @andsteing in https://github.com/google/flax/pull/2071 +- Add lifted conditional `nn.cond`. +- Improved error messages: parameters not found, loading checkpoints. +- Replace `jax.tree_multimap` (deprecated) with `jax.tree_map`. +- Add the "Module Lifecycle" design note. +- Add support for JAX dynamic stack-based named_call + +Bug fixes: +- Handle rate==1.0 edgecase in Dropout. +- Fix bug where Linen Module state is reused. +- Bug fixes and generalizations of nn.partitioning API. + +Breaking changes: +- 0.4.1