v0.3.2

Fixed an overly aggressive flax.nn deprecation warning in v0.3.1 that fired whenever you imported flax.

v0.3.1

Many improvements to Linen, and the old flax.nn is officially reprecated!

Notably, there's a clean API for extracting intermediates from modules
defined using @nn.compact, a more ergonomic API for using Batch Norm and Dropout in modules
defined using setup, support for MultiOptimizer with Linen, and multiple safety, performance
and error message improvements.

See the CHANGELOG for more details

Version 0.3.0

See changelog, overall linen API improvements as well as a few bug fixes.

Version 0.2.2

Various bug fixes and new features, including the Linen API, a new functional core and many Linen examples.

Version 0.2

Minor update to push some fixes to pypi.

version 0.1.0 rc2

v0.1.0rc2

bump version