Optax 0.1.8

What's Changed

• Remove legacy symbols. by @copybara-service in #397
• Move second order utilities to dedicated sub-package. by @copybara-service in #588
• Move stochastic gradient estimation functions to separate monte_carlo subpackage. by @copybara-service in #589
• Move losses to dedicated sub-package. by @copybara-service in #590
• Note TF's adam implementation difference by @vwxyzjn in #572
• Note RMSProp PyTorch vs Optax implementation difference by @vwxyzjn in #595
• Move complex_valued from experimental/ to contrib/. by @copybara-service in #602
• Make radam docstring point to radam paper. by @carlosgmartin in #510
• COntinuous COin Betting algorithm by @albcab in #536
• Update second_order package internal structure. by @copybara-service in #625
• Update github URL + nicer console highlighting by @copybara-service in #628
• Add import stub. by @copybara-service in #633
• Start optax.projections subpackage and add projection_non_negative. by @copybara-service in #632
• Also apply masking to compatible extra_args. by @copybara-service in #620
• Add tree_vdot to tree_utils. by @copybara-service in #634
• Add: merge multistep and apply_every logic by @celiolarcher in #596
• Remove redundant requirements folder by @copybara-service in #643
• Run pytype on Python 3.11 by @fabianp in #657
• Move MLP MNIST example to Colab notebook by @copybara-service in #658
• Move DP-SGD example to Colab by @copybara-service in #660
• Add tensorflow as dependency for the examples by @copybara-service in #661
• Add tensorflow-datasets to doc requirements by @copybara-service in #662
• D-Adaptation and Prodigy contrib implementations by @adefazio in #651
• Fix typing issue in prodigy.py for python 3.9 by @copybara-service in #671
• Fix CIFAR10 Flax&Optax example name in the gallery by @copybara-service in #672
• Move flax example to Colab by @copybara-service in #670
• update citation url since http://github.com/deepmind is now empty by @copybara-service in #667
• Move Haiku example to Colab notebook by @copybara-service in #673
• run pytype also on the contrib directory by @copybara-service in #675
• Add reduce_on_plateau LR scheduler to contrib directory. by @vz415 in #629
• Add to .gitignore temporary directories of building the docs. by @fabianp in #680
• Add the SAM example to the example gallery by @fabianp in #681
• Add reduce_on_plateau example (#679) by @copybara-service in #683
• Wrong image path on redue on plateau example by @copybara-service in #686
• Remove execution timeout (#687) by @copybara-service in #689
• Documentation Typo Fixes for apply_if_finite by @mharradon in #696
• FIX typos in public readme by @copybara-service in #691
• Add adversarial training example by @mmhamdy in #682
• Divide API documentation by @MaanasArora in #688
• Avoid scalar conversion of non-scalar arrays by @copybara-service in #693
• Improving docstrings for schedules. by @copybara-service in #692
• Fix TypeError in contrib.reduce_on_plateau() when x64 is enabled by @stefanocortinovis in #697
• Fix some doc links in README.md and other files #665 by @Bigpet in #706
• replace branch master by main by @copybara-service in #709
• New optax release by @copybara-service in #701

New Contributors

• @vwxyzjn made their first contribution in #572
• @carlosgmartin made their first contribution in #510
• @albcab made their first contribution in #536
• @celiolarcher made their first contribution in #596
• @adefazio made their first contribution in #651
• @vz415 made their first contribution in #629
• @mharradon made their first contribution in #696
• @mmhamdy made their first contribution in #682
• @MaanasArora made their first contribution in #688
• @stefanocortinovis made their first contribution in #697
• @Bigpet made their first contribution in #706

Full Changelog: v0.1.7...v0.1.8