Version 0.4.2
What's Changed
- Canonicalize conv padding by @jheek in #2009
- Update ScopeParamNotFoundError message. by @melissatan in #2013
- Set field on dataclass transform decorator by @NeilGirdhar in #1927
- Don't recommend mixing setup and compact in docs. by @levskaya in #2018
- Clarifies
optim.Adam(weight_decay)
parameter. by @copybara-service in #2016 - Update linear regression example in Jax intro and Flax intro. by @melissatan in #2015
- Lifted cond by @jheek in #2020
- Use tree_map instead of deprecated tree_multimap by @jheek in #2024
- Remove tree_multimap from docs, examples, and tests by @jheek in #2026
- Fix bug where the linen Module state is reused. by @jheek in #2025
- Add getattribute with lazy setup trigger. by @levskaya in #2028
- Better error messages for loading checkpoints. by @copybara-service in #2035
- Add filterwarning for jax.tree_multimap by @marcvanzee in #2038
- Adds Flax logo to README by @marcvanzee in #2036
- Module lifecycle note by @jheek in #1964
- Fix linter errors in core/scope.py and core/tracers.py. by @copybara-service in #2004
- Handle edge-case of rate==1.0 in Dropout layer. by @levskaya in #2055
- Bug fixes and generalizations of nn.partitioning api. by @copybara-service in #2062
- Add support for JAX dynamic stack-based named_call. by @copybara-service in #2063
- Updates pooling docstrings by @marcvanzee in #2064
- Makes annotated_mnist use Optax's xent loss. by @andsteing in #2071
Full Changelog: v0.4.1...v0.4.2