Skip to content

Commit

Permalink
Bumps Flax to v0.4.2 and adds CHANGELOG notes
Browse files Browse the repository at this point in the history
  • Loading branch information
marcvanzee committed May 5, 2022
1 parent 0d673cb commit 077a165
Show file tree
Hide file tree
Showing 3 changed files with 39 additions and 4 deletions.
39 changes: 37 additions & 2 deletions CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -8,15 +8,50 @@ vNext
-
-
-
- Added lifted conditional `nn.cond`
- Handle rate==1.0 edgecase in Dropout.
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-

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
-----
Expand Down
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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},
}
```
Expand Down
2 changes: 1 addition & 1 deletion flax/version.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,5 +13,5 @@
# limitations under the License.

"""Current Flax version at head on Github."""
__version__ = "0.4.1"
__version__ = "0.4.2"

0 comments on commit 077a165

Please sign in to comment.