v0.2.0

What's Changed

• Begin of 0.2.0 development by @copybara-service in #766
• Remove unnecessary %pip install statements from the notebooks by @JayJayleee in #762
• Updates warmup_cosine_decay_schedule to allow 0 as peak_value. Currently it errors out as divide by 0. by @copybara-service in #770
• fixed math flex direction by @copybara-service in #775
• Docstrings for all solvers with Objective Function values by @copybara-service in #771
• Misc improvements in adversarial_training.ipynb by @copybara-service in #776
• Solver API (wip) by @copybara-service in #777
• Detail that learning_rate can be scalar or a schedule. by @copybara-service in #778
• workaround for pylint issues by @fabianp in #783
• correct type annotation for params by @yixiaoer in #785
• Ensure no execution of slow notebooks. by @copybara-service in #788
• [DOC] merge development and design_docs into a single file by @fabianp in #789
• Add development information in README. by @mblondel in #790
• Solve #767, some examples not being displayed on the webpage by @fabianp in #791
• Add missing examples to contrib examples section. by @fabianp in #794
• Make EmptyState a valid ArrayTree by @copybara-service in #752
• Improve error message for cosine_decay_schedule. by @copybara-service in #797
• move DPSGD docs to contrib/ section by @WangHanSolo in #782
• Adjusted 'test.sh' for universal macOS and Linux Compatibility by @hmludwig in #804
• Benchmark recommendations. by @copybara-service in #807
• Removed 'examples/quick_start.ipynb' and adjusted references. by @hmludwig in #811
• [DOC] streamline sidebar of gallery by @fabianp in #810
• [FIX] bad indentation in the docstring of nadam by @fabianp in #812
• Backtracking linesearch. by @copybara-service in #795
• Fixing backtracking linesearch support on gpus by @copybara-service in #814
• allow wrappers.flatten to work with 0-dim arrays by @copybara-service in #816
• Add Example, Args and Returns section to optax.linear_schedule by @copybara-service in #818
• FIX doctest format in TransformUpdateExtraArgsFn by @copybara-service in #820
• Refactoring and link fixes in development.md by @fabianp in #822
• Clarify formula for cosine learning rate. by @copybara-service in #823
• Detail SGD with momentum equations. by @copybara-service in #830
• Fixed the rendering of equations in jupyter notebooks by @hmludwig in #836
• Fixed the Sharpness aware minimization docstring by @hmludwig in #834
• FIX "gallery.rst:5: WARNING: Title underline too short." by @copybara-service in #838
• Moved linear algebra operations into utilities by @hmludwig in #837
• adding tree utils for generating trees with random values and averaging across trees. by @copybara-service in #825
• v0.2.0 release by @copybara-service in #844

New Contributors

• @JayJayleee made their first contribution in #762
• @yixiaoer made their first contribution in #785
• @mblondel made their first contribution in #790
• @WangHanSolo made their first contribution in #782
• @hmludwig made their first contribution in #804

Full Changelog: v0.1.9...v0.2.0