Version 0.4.0

What's Changed

• Add PReLU Activation by @isaaccorley in #1570
• Fix GroupNorm type hint for param num_groups. by @lkhphuc in #1657
• Add named_call overrides to docs by @jheek in #1649
• mission statement by @jheek in #1668
• Improves Flax Modules for RTD by @marcvanzee in #1416
• Add clarifying docstring for 'size' argument to prefetch_to_device's by @avital in #1574
• Add circular padding to flax.linen.Conv and flax.linen.ConvTranspose by @sgrigory in #1661
• Fix child scope rng reuse. by @jheek in #1692
• Numerically stable weight norm by @jheek in #1693
• Remove cyclic refs from scope by @jheek in #1696
• Add unroll to jax_utils.scan_in_dim by @ptigwe in #1691
• Removes rng arguments from Dropout's __call__. by @copybara-service in #1689
• Add error for empty scopes. by @jheek in #1698
• correct axis resolution in case of repeated axis in the logica axis r… by @ultrons in #1703
• Fix lost mutation bug in transforms on nested scopes. by @levskaya in #1716
• Expose put_variable function to Module. by @levskaya in #1710
• add eq and hash for scopes by @jheek in #1720
• Fixes a bug in DenseGeneral. by @copybara-service in #1722
• Add param_dtype argument to linen Modules by @jheek in #1739
• Implement custom vjp by @jheek in #1738
• Handle setup with transformed methods taking submodules of self. by @levskaya in #1745
• validate RNG key shape against jax's default by @copybara-service in #1780
• Adds optax update guide. by @andsteing in #1774
• Implement LazyRNG by @jheek in #1723
• make params_with_axes() work when params_axes is not mutable by @copybara-service in #1811
• Updates the ensembling HOWTO to Optax. by @andsteing in #1806
• Adds prominent scenic link to examples/README.md by @copybara-service in #1809
• Removes PixelCNN++ example. @copybara-service in #1819
• Add support for non-float32 normalization for linen normalization layers by @jheek in #1804
• Make Filter a Collection instead of a Container by @NeilGirdhar in #1815
• Removes deprecated API from RTD by @marcvanzee in #1824

New Contributors

• @isaaccorley made their first contribution in #1570
• @lkhphuc made their first contribution in #1657
• @sgrigory made their first contribution in #1661
• @ptigwe made their first contribution in #1691
• @ultrons made their first contribution in #1703
• @dependabot made their first contribution in #1749
• @NeilGirdhar made their first contribution in #1699
• @saeta made their first contribution in #1784
• @melissatan made their first contribution in #1793

Full Changelog: v0.3.6...v0.4.0