Skip to content

jaxopt-v0.3.1
Compare
Choose a tag to compare
@mblondel mblondel released this 28 Feb 21:55
· 387 commits to main since this release
jaxopt-v0.3.1
d35aac3

New features

  • Pjit-based example of data parallel training using Flax, by Felipe Llinares.

Bug fixes and enhancements

  • Support for GPU and state of the art adversarial training algorithm (PGD) on the robust_training.py example, by Fabian Pedregosa.
  • Update line search in LBFGS to use jit and unroll from LBFGS, by Ian Williamson.
  • Support dynamic maximum iteration count in iterative solvers, by Roy Frostig.
  • Fix tree_where for singleton pytrees, by Louis Béthune.
  • Remove QuadraticProg in projections and set init_params=None by default in QP solvers, by Louis Béthune.
  • Add missing 'value' attribute in LbfgsState, by Mathieu Blondel.

Contributors

Felipe Llinares, Fabian Pedregosa, Ian Williamson, Louis Bétune, Mathieu Blondel, Roy Frostig.

Assets 2
Loading