diff --git a/CHANGELOG.md b/CHANGELOG.md index bdb625c280..03715e8513 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -18,12 +18,26 @@ vNext - - - -- Add copy() method to Module. This is a user-friendly version of the internal clone() method with better - defaults for common use cases. - - - -- NOTE: Remember to bump version number to 0.8.0 +- +- + +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. +- Added `nn.compact_name_scope` decorator that enables methods to act as compact name scopes as with regular Haiku methods. This makes porting Haiku code easier. +- Add copy() method to Module. This is a user-friendly version of the internal clone() method with better + defaults for common use cases. +- Added [`BatchApply`](https://flax.readthedocs.io/en/latest/api_reference/flax.linen/layers.html#batchapply) class. +- Added `sow_weights` option in attention layer. +- Added [`MultiHeadAttention`](https://flax.readthedocs.io/en/latest/api_reference/flax.linen/_autosummary/flax.linen.MultiHeadAttention.html) alias. +- Added kwargs support for `nn.jit`. +- Deprecated `normalize` activation function, in favor of `standardize`. +- Added `GeGLU` activation function. +- Added `Enum` support for `tabulate` function. +- Added simple argument-only lifted `nn.grad` function. 0.7.5 ----- diff --git a/README.md b/README.md index 750d333958..c2c1b851ce 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.5}, + version = {0.8.0}, year = {2023}, } ```