JAXopt

Documentation

  • Basics
  • Unconstrained optimization
  • Constrained optimization
  • Quadratic programming
  • Non-smooth optimization
  • Stochastic optimization
  • Root finding
  • Fixed point resolution
  • Nonlinear least squares
  • Linear system solving
  • Implicit differentiation
  • Loss and objective functions
  • Line search
  • Perturbed optimization

API

  • API at a glance

Examples

  • Notebook gallery
  • Example gallery

About

  • Authors
  • Changelog
  • Source code
  • Issue tracker
  • Development
JAXopt
  • »
  • Overview: module code

All modules for which code is available

  • jaxopt._src.anderson
  • jaxopt._src.anderson_wrapper
  • jaxopt._src.armijo_sgd
  • jaxopt._src.backtracking_linesearch
  • jaxopt._src.bfgs
  • jaxopt._src.bisection
  • jaxopt._src.block_cd
  • jaxopt._src.broyden
  • jaxopt._src.cd_qp
  • jaxopt._src.cvxpy_wrapper
  • jaxopt._src.eq_qp
  • jaxopt._src.fixed_point_iteration
  • jaxopt._src.gauss_newton
  • jaxopt._src.gradient_descent
  • jaxopt._src.hager_zhang_linesearch
  • jaxopt._src.implicit_diff
  • jaxopt._src.isotonic
  • jaxopt._src.iterative_refinement
  • jaxopt._src.lbfgs
  • jaxopt._src.lbfgsb
  • jaxopt._src.levenberg_marquardt
  • jaxopt._src.linear_solve
  • jaxopt._src.loss
  • jaxopt._src.mirror_descent
  • jaxopt._src.nonlinear_cg
  • jaxopt._src.objective
  • jaxopt._src.optax_wrapper
  • jaxopt._src.osqp
  • jaxopt._src.perturbations
  • jaxopt._src.polyak_sgd
  • jaxopt._src.projected_gradient
  • jaxopt._src.projection
  • jaxopt._src.prox
  • jaxopt._src.proximal_gradient
  • jaxopt._src.scipy_wrappers
  • jaxopt._src.tree_util

© Copyright 2021-2022, the JAXopt authors.

Built with Sphinx using a theme provided by Read the Docs.