Version 0.4.3

[marcvanzee]

What's Changed

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

Full Changelog: v0.4.1...v0.4.2

---

This discussion was created from the release Version 0.4.2.