Version 0.5.2

What's Changed

• Flax Basics docs: Add missing @jax.jit to mse by @rsokl in #2181
• add missing colon in example code by @PWhiddy in #2188
• New-sphinx-theme by @cgarciae in #2171
• Add missing PyYAML dependency by @cgarciae in #2193
• Improve module docs by @cgarciae in #2167
• Changed optimizer to optax by @berndbohnet in #1916
• Show repository button by @PhilipVinc in #2206
• Updates filterwarning in pytest.ini by @marcvanzee in #2209
• v0.5.2 by @cgarciae in #2203

New Contributors

• @rsokl made their first contribution in #2181
• @PWhiddy made their first contribution in #2188
• @berndbohnet made their first contribution in #1916

Full Changelog: v0.5.1...v0.5.2