Skip to content

jaxopt-v0.5.5
Compare
Choose a tag to compare
@mblondel mblondel released this 20 Oct 09:08
· 225 commits to main since this release
jaxopt-v0.5.5
2ff6fc7

New features

  • Added MAML example by Fabian Pedregosa based on initial code by Paul Vicol and Eric Jiang.
  • Added the possibility to stop LBFGS after a line search failure, by Zaccharie Ramzi.
  • Added gamma to LBFGS state, by Zaccharie Ramzi.
  • Added jaxopt.BFGS, by Mathieu Blondel.
  • Added value_and_grad option to all gradient-based solvers, by Mathieu Blondel.
  • Added Fenchel-Young loss, by Quentin Berthet.
  • Added projection_sparse_simplex, by Tianlin Liu.

Bug fixes and enhancements

  • Fixed missing args,kwargs in resnet example, by Louis Béthune.
  • Corrected the implicit diff examples, by Zaccharie Ramzi.
  • Small optimization in l2-regularized semi-dual OT, by Mathieu Blondel.
  • Numerical stability improvements in jaxopt.LevenbergMarquardt, by Amir Saadat.
  • Dtype consistency in LBFGS, by Alex Botev.

Deprecations

  • jaxopt.QuadraticProgramming is now fully removed. Use jaxopt.CvxpyQP, jaxopt.OSQP, jaxopt.BoxOSQP and jaxopt.EqualityConstrainedQP instead.

Contributors

Alex Botev, Amir Saadat, Fabian Pedregosa, Louis Béthune, Mathieu Blondel, Quentin Berthet, Tianlin Liu, Zaccharie Ramzi.

Assets 2
Loading