Skip to content

Version 0.4.2

Compare
Choose a tag to compare
@marcvanzee marcvanzee released this 05 May 12:37
· 2181 commits to main since this release

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