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
TypeErrorincontrib.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