github google-deepmind/optax v0.2.0

latest releases: v0.2.2, v0.2.1
2 months ago

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

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

Don't miss a new optax release

NewReleases is sending notifications on new releases.