Skip to content

jaxopt-v0.1
Compare
Choose a tag to compare
@mblondel mblondel released this 14 Oct 11:28
· 533 commits to main since this release
jaxopt-v0.1
70c80d1

Classes

  • jaxopt.AndersonAcceleration
  • jaxopt.AndersonWrapper
  • jaxopt.Bisection
  • jaxopt.BlockCoordinateDescent
  • jaxopt.FixedPointIteration
  • jaxopt.GradientDescent
  • jaxopt.MirrorDescent
  • jaxopt.OptaxSolver
  • jaxopt.PolyakSGD
  • jaxopt.ProjectedGradient
  • jaxopt.ProximalGradient
  • jaxopt.QuadraticProgramming
  • jaxopt.ScipyBoundedLeastSquares
  • jaxopt.ScipyBoundedMinimize
  • jaxopt.ScipyLeastSquares
  • jaxopt.ScipyMinimize
  • jaxopt.ScipyRootFinding
  • Implicit differentiation

Examples

  • Binary kernel SVM with intercept.
  • Image classification example with Flax and JAXopt.
  • Image classification example with Haiku and JAXopt.
  • VAE example with Haiku and JAXopt.
  • Implicit differentiation of lasso.
  • Multiclass linear SVM (without intercept).
  • Non-negative matrix factorizaton (NMF) using alternating minimization.
  • Dataset distillation.
  • Implicit differentiation of ridge regression.
  • Robust training.
  • Anderson acceleration of gradient descent.
  • Anderson acceleration of block coordinate descent.
  • Anderson acceleration in application to Picard–Lindelöf theorem.

Contributors

Fabian Pedregosa, Felipe Llinares, Robert Gower, Louis Bethune, Marco Cuturi, Mathieu Blondel, Peter Hawkins, Quentin Berthet, Roy Frostig, Ta-Chu Kao

Assets 2
Loading