Index

_

__init__() (jaxopt.AndersonAcceleration method) (jaxopt.AndersonWrapper method) (jaxopt.ArmijoSGD method) (jaxopt.BacktrackingLineSearch method) (jaxopt.BFGS method) (jaxopt.Bisection method) (jaxopt.BlockCoordinateDescent method) (jaxopt.BoxCDQP method) (jaxopt.BoxOSQP method) (jaxopt.Broyden method) (jaxopt.CvxpyQP method) (jaxopt.EqualityConstrainedQP method) (jaxopt.FixedPointIteration method) (jaxopt.GaussNewton method) (jaxopt.GradientDescent method) (jaxopt.HagerZhangLineSearch method) (jaxopt.IterativeRefinement method) (jaxopt.LBFGS method) (jaxopt.LBFGSB method) (jaxopt.LevenbergMarquardt method) (jaxopt.MirrorDescent method) (jaxopt.NonlinearCG method) (jaxopt.OptaxSolver method) (jaxopt.OSQP method) (jaxopt.perturbations.Gumbel method) (jaxopt.perturbations.Normal method) (jaxopt.PolyakSGD method) (jaxopt.ProjectedGradient method) (jaxopt.ProximalGradient method) (jaxopt.ScipyBoundedMinimize method) (jaxopt.ScipyMinimize method) (jaxopt.ScipyRootFinding method)

A

acceleration (jaxopt.GradientDescent attribute) (jaxopt.ProjectedGradient attribute) (jaxopt.ProximalGradient attribute)

aggressiveness (jaxopt.ArmijoSGD attribute) AndersonAcceleration (class in jaxopt) AndersonWrapper (class in jaxopt) ArmijoSGD (class in jaxopt)

B

BacktrackingLineSearch (class in jaxopt) beta (jaxopt.AndersonAcceleration attribute) (jaxopt.AndersonWrapper attribute) BFGS (class in jaxopt) binary_hinge_loss() (in module jaxopt.loss) binary_logistic_loss() (in module jaxopt.loss) binary_logreg (in module jaxopt.objective)

binary_perceptron_loss() (in module jaxopt.loss) binary_sparsemax_loss() (in module jaxopt.loss) Bisection (class in jaxopt) block_prox (jaxopt.BlockCoordinateDescent attribute) BlockCoordinateDescent (class in jaxopt) BoxCDQP (class in jaxopt) BoxOSQP (class in jaxopt) Broyden (class in jaxopt)

C

c1 (jaxopt.BacktrackingLineSearch attribute) (jaxopt.HagerZhangLineSearch attribute) c2 (jaxopt.BacktrackingLineSearch attribute) (jaxopt.HagerZhangLineSearch attribute) callback (jaxopt.ScipyMinimize attribute) check_bracket (jaxopt.Bisection attribute) check_primal_dual_infeasability (jaxopt.BoxOSQP attribute) (jaxopt.OSQP attribute)

condition (jaxopt.BacktrackingLineSearch attribute) (jaxopt.BFGS attribute) (jaxopt.LBFGS attribute) (jaxopt.LBFGSB attribute) (jaxopt.NonlinearCG attribute) contribution_ratio_threshold (jaxopt.LevenbergMarquardt attribute) custom_fixed_point() (in module jaxopt.implicit_diff) custom_root() (in module jaxopt.implicit_diff) CvxpyQP (class in jaxopt)

D

damping_parameter (jaxopt.LevenbergMarquardt attribute) decrease_factor (jaxopt.ArmijoSGD attribute) (jaxopt.BacktrackingLineSearch attribute) (jaxopt.BFGS attribute) (jaxopt.Broyden attribute) (jaxopt.LBFGS attribute) (jaxopt.LBFGSB attribute) (jaxopt.NonlinearCG attribute) (jaxopt.ProximalGradient attribute)

delta (jaxopt.PolyakSGD attribute) dtype (jaxopt.ScipyBoundedMinimize attribute), [1] (jaxopt.ScipyMinimize attribute), [1] (jaxopt.ScipyRootFinding attribute), [1] dual_infeasible_tol (jaxopt.BoxOSQP attribute) (jaxopt.OSQP attribute)

E

eq_qp_solve (jaxopt.BoxOSQP attribute) (jaxopt.OSQP attribute)

EqualityConstrainedQP (class in jaxopt)

F

fixed_point_fun (jaxopt.AndersonAcceleration attribute) FixedPointIteration (class in jaxopt) fun (jaxopt.ArmijoSGD attribute) (jaxopt.BacktrackingLineSearch attribute) (jaxopt.BFGS attribute) (jaxopt.BlockCoordinateDescent attribute) (jaxopt.BoxOSQP attribute) (jaxopt.Broyden attribute) (jaxopt.EqualityConstrainedQP attribute) (jaxopt.GradientDescent attribute) (jaxopt.HagerZhangLineSearch attribute) (jaxopt.LBFGS attribute) (jaxopt.LBFGSB attribute) (jaxopt.MirrorDescent attribute) (jaxopt.NonlinearCG attribute) (jaxopt.OptaxSolver attribute) (jaxopt.OSQP attribute) (jaxopt.PolyakSGD attribute) (jaxopt.ProjectedGradient attribute) (jaxopt.ProximalGradient attribute) (jaxopt.ScipyBoundedMinimize attribute) (jaxopt.ScipyMinimize attribute)

G

gamma (jaxopt.Broyden attribute) GaussNewton (class in jaxopt) geodesic (jaxopt.LevenbergMarquardt attribute)

GradientDescent (class in jaxopt) gtol (jaxopt.LevenbergMarquardt attribute) Gumbel (class in jaxopt.perturbations)

H

HagerZhangLineSearch (class in jaxopt) has_aux (jaxopt.AndersonAcceleration attribute) (jaxopt.ArmijoSGD attribute) (jaxopt.BacktrackingLineSearch attribute) (jaxopt.BFGS attribute) (jaxopt.Broyden attribute) (jaxopt.FixedPointIteration attribute) (jaxopt.GradientDescent attribute) (jaxopt.HagerZhangLineSearch attribute) (jaxopt.LBFGS attribute) (jaxopt.LBFGSB attribute) (jaxopt.MirrorDescent attribute) (jaxopt.NonlinearCG attribute) (jaxopt.OptaxSolver attribute) (jaxopt.PolyakSGD attribute) (jaxopt.ProjectedGradient attribute), [1] (jaxopt.ProximalGradient attribute) (jaxopt.ScipyBoundedMinimize attribute) (jaxopt.ScipyMinimize attribute) (jaxopt.ScipyRootFinding attribute)

history_size (jaxopt.AndersonAcceleration attribute) (jaxopt.AndersonWrapper attribute) (jaxopt.Broyden attribute) (jaxopt.LBFGS attribute) (jaxopt.LBFGSB attribute) huber_loss() (in module jaxopt.loss)

I

implicit_diff (jaxopt.AndersonAcceleration attribute) (jaxopt.AndersonWrapper attribute) (jaxopt.ArmijoSGD attribute) (jaxopt.BFGS attribute) (jaxopt.BlockCoordinateDescent attribute) (jaxopt.BoxCDQP attribute) (jaxopt.BoxOSQP attribute) (jaxopt.Broyden attribute) (jaxopt.EqualityConstrainedQP attribute) (jaxopt.FixedPointIteration attribute) (jaxopt.GaussNewton attribute) (jaxopt.GradientDescent attribute) (jaxopt.IterativeRefinement attribute) (jaxopt.LBFGS attribute) (jaxopt.LBFGSB attribute) (jaxopt.LevenbergMarquardt attribute) (jaxopt.MirrorDescent attribute) (jaxopt.NonlinearCG attribute) (jaxopt.OptaxSolver attribute) (jaxopt.PolyakSGD attribute) (jaxopt.ProjectedGradient attribute) (jaxopt.ProximalGradient attribute) implicit_diff_solve (jaxopt.AndersonAcceleration attribute) (jaxopt.AndersonWrapper attribute) (jaxopt.ArmijoSGD attribute) (jaxopt.BFGS attribute) (jaxopt.Bisection attribute) (jaxopt.BlockCoordinateDescent attribute) (jaxopt.BoxCDQP attribute) (jaxopt.BoxOSQP attribute) (jaxopt.Broyden attribute) (jaxopt.EqualityConstrainedQP attribute) (jaxopt.FixedPointIteration attribute) (jaxopt.GaussNewton attribute) (jaxopt.GradientDescent attribute) (jaxopt.IterativeRefinement attribute) (jaxopt.LBFGS attribute) (jaxopt.LBFGSB attribute) (jaxopt.LevenbergMarquardt attribute) (jaxopt.MirrorDescent attribute) (jaxopt.NonlinearCG attribute) (jaxopt.OptaxSolver attribute) (jaxopt.OSQP attribute) (jaxopt.PolyakSGD attribute) (jaxopt.ProjectedGradient attribute) (jaxopt.ProximalGradient attribute) (jaxopt.ScipyBoundedMinimize attribute) (jaxopt.ScipyMinimize attribute) (jaxopt.ScipyRootFinding attribute)

increase_factor (jaxopt.ArmijoSGD attribute) (jaxopt.BFGS attribute) (jaxopt.Broyden attribute) (jaxopt.LBFGS attribute) (jaxopt.LBFGSB attribute) (jaxopt.NonlinearCG attribute) init_params() (jaxopt.BoxOSQP method) (jaxopt.OSQP method) init_state() (jaxopt.AndersonAcceleration method) (jaxopt.ArmijoSGD method) (jaxopt.BacktrackingLineSearch method) (jaxopt.BFGS method) (jaxopt.Bisection method) (jaxopt.BlockCoordinateDescent method) (jaxopt.BoxCDQP method) (jaxopt.BoxOSQP method) (jaxopt.Broyden method) (jaxopt.FixedPointIteration method) (jaxopt.GaussNewton method) (jaxopt.GradientDescent method) (jaxopt.HagerZhangLineSearch method) (jaxopt.LBFGS method) (jaxopt.LBFGSB method) (jaxopt.LevenbergMarquardt method) (jaxopt.MirrorDescent method) (jaxopt.NonlinearCG method) (jaxopt.OptaxSolver method) (jaxopt.PolyakSGD method) (jaxopt.ProjectedGradient method) (jaxopt.ProximalGradient method) isotonic_l2_pav() (in module jaxopt.isotonic) IterativeRefinement (class in jaxopt)

J

jit (jaxopt.AndersonAcceleration attribute) (jaxopt.AndersonWrapper attribute) (jaxopt.ArmijoSGD attribute) (jaxopt.BacktrackingLineSearch attribute) (jaxopt.BFGS attribute) (jaxopt.Bisection attribute) (jaxopt.BlockCoordinateDescent attribute) (jaxopt.BoxCDQP attribute) (jaxopt.BoxOSQP attribute) (jaxopt.Broyden attribute) (jaxopt.EqualityConstrainedQP attribute) (jaxopt.FixedPointIteration attribute) (jaxopt.GaussNewton attribute) (jaxopt.HagerZhangLineSearch attribute) (jaxopt.IterativeRefinement attribute) (jaxopt.LBFGS attribute) (jaxopt.LBFGSB attribute) (jaxopt.LevenbergMarquardt attribute) (jaxopt.MirrorDescent attribute) (jaxopt.NonlinearCG attribute) (jaxopt.OptaxSolver attribute) (jaxopt.PolyakSGD attribute) (jaxopt.ProjectedGradient attribute) (jaxopt.ProximalGradient attribute) (jaxopt.ScipyBoundedMinimize attribute) (jaxopt.ScipyMinimize attribute) (jaxopt.ScipyRootFinding attribute)

K

kl_projection_birkhoff() (in module jaxopt.projection)

kl_projection_transport() (in module jaxopt.projection)

L

l2_multiclass_logreg() (in module jaxopt.objective) l2_multiclass_logreg_with_intercept() (in module jaxopt.objective) l2_optimality_error() (jaxopt.AndersonAcceleration method) (jaxopt.AndersonWrapper method) (jaxopt.ArmijoSGD method) (jaxopt.BacktrackingLineSearch method) (jaxopt.BFGS method) (jaxopt.Bisection method) (jaxopt.BlockCoordinateDescent method) (jaxopt.BoxCDQP method) (jaxopt.BoxOSQP method) (jaxopt.Broyden method) (jaxopt.CvxpyQP method) (jaxopt.EqualityConstrainedQP method) (jaxopt.FixedPointIteration method) (jaxopt.GaussNewton method) (jaxopt.GradientDescent method) (jaxopt.HagerZhangLineSearch method) (jaxopt.IterativeRefinement method) (jaxopt.LBFGS method) (jaxopt.LBFGSB method) (jaxopt.LevenbergMarquardt method) (jaxopt.MirrorDescent method) (jaxopt.NonlinearCG method) (jaxopt.OptaxSolver method) (jaxopt.OSQP method) (jaxopt.PolyakSGD method) (jaxopt.ProjectedGradient method) (jaxopt.ProximalGradient method) (jaxopt.ScipyBoundedMinimize method) (jaxopt.ScipyMinimize method) (jaxopt.ScipyRootFinding method)

LBFGS (class in jaxopt) LBFGSB (class in jaxopt) least_squares (in module jaxopt.objective) LevenbergMarquardt (class in jaxopt) linesearch (jaxopt.BFGS attribute) (jaxopt.Broyden attribute) (jaxopt.LBFGS attribute) (jaxopt.NonlinearCG attribute) linesearch_init (jaxopt.BFGS attribute) (jaxopt.LBFGS attribute) (jaxopt.LBFGSB attribute) (jaxopt.NonlinearCG attribute) lower (jaxopt.Bisection attribute)

M

make_perturbed_argmax() (in module jaxopt.perturbations) make_perturbed_fun() (in module jaxopt.perturbations) make_perturbed_max() (in module jaxopt.perturbations) make_projection_grad() (jaxopt.MirrorDescent static method) make_prox_from_projection() (in module jaxopt.prox) matvec_A (jaxopt.BoxOSQP attribute) (jaxopt.EqualityConstrainedQP attribute) (jaxopt.IterativeRefinement attribute) (jaxopt.OSQP attribute) matvec_A_bar (jaxopt.IterativeRefinement attribute) matvec_G (jaxopt.OSQP attribute) matvec_Q (jaxopt.BoxOSQP attribute) (jaxopt.EqualityConstrainedQP attribute) (jaxopt.OSQP attribute) max_stepsize (jaxopt.ArmijoSGD attribute) (jaxopt.BacktrackingLineSearch attribute) (jaxopt.BFGS attribute) (jaxopt.Broyden attribute) (jaxopt.HagerZhangLineSearch attribute) (jaxopt.LBFGS attribute) (jaxopt.LBFGSB attribute) (jaxopt.NonlinearCG attribute) (jaxopt.PolyakSGD attribute) maxiter (jaxopt.AndersonAcceleration attribute) (jaxopt.ArmijoSGD attribute) (jaxopt.BacktrackingLineSearch attribute) (jaxopt.BFGS attribute) (jaxopt.Bisection attribute) (jaxopt.BlockCoordinateDescent attribute) (jaxopt.BoxCDQP attribute) (jaxopt.BoxOSQP attribute) (jaxopt.Broyden attribute) (jaxopt.EqualityConstrainedQP attribute) (jaxopt.FixedPointIteration attribute) (jaxopt.GaussNewton attribute) (jaxopt.GradientDescent attribute) (jaxopt.HagerZhangLineSearch attribute) (jaxopt.IterativeRefinement attribute) (jaxopt.LBFGS attribute) (jaxopt.LBFGSB attribute) (jaxopt.LevenbergMarquardt attribute) (jaxopt.MirrorDescent attribute) (jaxopt.NonlinearCG attribute) (jaxopt.OptaxSolver attribute) (jaxopt.OSQP attribute) (jaxopt.PolyakSGD attribute) (jaxopt.ProjectedGradient attribute) (jaxopt.ProximalGradient attribute)

maxls (jaxopt.ArmijoSGD attribute) (jaxopt.BFGS attribute) (jaxopt.Broyden attribute) (jaxopt.GradientDescent attribute) (jaxopt.LBFGS attribute) (jaxopt.LBFGSB attribute) (jaxopt.NonlinearCG attribute) (jaxopt.ProjectedGradient attribute) (jaxopt.ProximalGradient attribute) method (jaxopt.NonlinearCG attribute) (jaxopt.ScipyBoundedMinimize attribute) (jaxopt.ScipyMinimize attribute) (jaxopt.ScipyRootFinding attribute) min_stepsize (jaxopt.BFGS attribute) (jaxopt.Broyden attribute) (jaxopt.LBFGS attribute) (jaxopt.LBFGSB attribute) (jaxopt.NonlinearCG attribute) MirrorDescent (class in jaxopt) mixing_frequency (jaxopt.AndersonAcceleration attribute) (jaxopt.AndersonWrapper attribute) momentum (jaxopt.ArmijoSGD attribute) (jaxopt.BoxOSQP attribute) (jaxopt.OSQP attribute) (jaxopt.PolyakSGD attribute) multiclass_hinge_loss() (in module jaxopt.loss) multiclass_linear_svm_dual (in module jaxopt.objective) multiclass_logistic_loss() (in module jaxopt.loss) multiclass_logreg (in module jaxopt.objective) multiclass_logreg_with_intercept() (in module jaxopt.objective) multiclass_perceptron_loss() (in module jaxopt.loss) multiclass_sparsemax_loss() (in module jaxopt.loss)

N

NonlinearCG (class in jaxopt)

Normal (class in jaxopt.perturbations)

O

opt (jaxopt.OptaxSolver attribute) OptaxSolver (class in jaxopt) optimality_fun (jaxopt.Bisection attribute) (jaxopt.ScipyRootFinding attribute) optimality_fun() (jaxopt.AndersonAcceleration method) (jaxopt.AndersonWrapper method) (jaxopt.ArmijoSGD method) (jaxopt.BFGS method) (jaxopt.BlockCoordinateDescent method) (jaxopt.Broyden method) (jaxopt.FixedPointIteration method) (jaxopt.GradientDescent method) (jaxopt.LBFGS method) (jaxopt.LBFGSB method) (jaxopt.LevenbergMarquardt method) (jaxopt.MirrorDescent method) (jaxopt.NonlinearCG method) (jaxopt.OptaxSolver method) (jaxopt.PolyakSGD method) (jaxopt.ProjectedGradient method) (jaxopt.ProximalGradient method) (jaxopt.ScipyBoundedMinimize method) (jaxopt.ScipyMinimize method)

options (jaxopt.ScipyBoundedMinimize attribute) (jaxopt.ScipyMinimize attribute) (jaxopt.ScipyRootFinding attribute) OSQP (class in jaxopt)

P

PolyakSGD (class in jaxopt) pre_update (jaxopt.ArmijoSGD attribute) (jaxopt.OptaxSolver attribute) (jaxopt.PolyakSGD attribute) primal_infeasible_tol (jaxopt.BoxOSQP attribute) (jaxopt.OSQP attribute) ProjectedGradient (class in jaxopt) projection (jaxopt.ProjectedGradient attribute) projection_affine_set() (in module jaxopt.projection) projection_birkhoff() (in module jaxopt.projection) projection_box() (in module jaxopt.projection) projection_box_section() (in module jaxopt.projection) projection_grad (jaxopt.MirrorDescent attribute) projection_halfspace() (in module jaxopt.projection) projection_hyperplane() (in module jaxopt.projection) projection_l1_ball() (in module jaxopt.projection) projection_l1_sphere() (in module jaxopt.projection) projection_l2_ball() (in module jaxopt.projection)

projection_l2_sphere() (in module jaxopt.projection) projection_linf_ball() (in module jaxopt.projection) projection_non_negative() (in module jaxopt.projection) projection_polyhedron() (in module jaxopt.projection) projection_simplex() (in module jaxopt.projection) projection_sparse_simplex() (in module jaxopt.projection) projection_transport() (in module jaxopt.projection) prox (jaxopt.ProximalGradient attribute) prox() (jaxopt.GradientDescent method) (jaxopt.ProximalGradient method) prox_elastic_net() (in module jaxopt.prox) prox_group_lasso() (in module jaxopt.prox) prox_lasso() (in module jaxopt.prox) prox_non_negative_lasso() (in module jaxopt.prox) prox_non_negative_ridge() (in module jaxopt.prox) prox_none() (in module jaxopt.prox) prox_ridge() (in module jaxopt.prox) ProximalGradient (class in jaxopt)

R

refine_maxiter (jaxopt.EqualityConstrainedQP attribute) refine_regularization (jaxopt.EqualityConstrainedQP attribute) reset_option (jaxopt.ArmijoSGD attribute) reset_stepsize() (jaxopt.ArmijoSGD method) residual_fun (jaxopt.GaussNewton attribute) (jaxopt.LevenbergMarquardt attribute) rho_max (jaxopt.BoxOSQP attribute) (jaxopt.OSQP attribute) rho_min (jaxopt.BoxOSQP attribute) (jaxopt.OSQP attribute) rho_start (jaxopt.BoxOSQP attribute) (jaxopt.OSQP attribute) ridge (jaxopt.AndersonAcceleration attribute) (jaxopt.AndersonWrapper attribute) ridge_regression() (in module jaxopt.objective) root_jvp() (in module jaxopt.implicit_diff) root_vjp() (in module jaxopt.implicit_diff) run() (jaxopt.AndersonAcceleration method) (jaxopt.AndersonWrapper method) (jaxopt.ArmijoSGD method) (jaxopt.BacktrackingLineSearch method) (jaxopt.BFGS method) (jaxopt.Bisection method) (jaxopt.BlockCoordinateDescent method) (jaxopt.BoxCDQP method) (jaxopt.BoxOSQP method) (jaxopt.Broyden method) (jaxopt.CvxpyQP method) (jaxopt.EqualityConstrainedQP method) (jaxopt.FixedPointIteration method) (jaxopt.GaussNewton method) (jaxopt.GradientDescent method) (jaxopt.HagerZhangLineSearch method) (jaxopt.IterativeRefinement method) (jaxopt.LBFGS method) (jaxopt.LBFGSB method) (jaxopt.LevenbergMarquardt method) (jaxopt.MirrorDescent method) (jaxopt.NonlinearCG method) (jaxopt.OptaxSolver method) (jaxopt.OSQP method) (jaxopt.PolyakSGD method) (jaxopt.ProjectedGradient method) (jaxopt.ProximalGradient method) (jaxopt.ScipyBoundedMinimize method) (jaxopt.ScipyMinimize method) (jaxopt.ScipyRootFinding method)

run_iterator() (jaxopt.ArmijoSGD method) (jaxopt.OptaxSolver method) (jaxopt.PolyakSGD method)

S

ScipyBoundedMinimize (class in jaxopt) ScipyMinimize (class in jaxopt) ScipyRootFinding (class in jaxopt) sigma (jaxopt.BoxOSQP attribute) (jaxopt.OSQP attribute) solve (jaxopt.EqualityConstrainedQP attribute) (jaxopt.IterativeRefinement attribute) solve() (jaxopt.EqualityConstrainedQP method) (jaxopt.IterativeRefinement method) solve_bicgstab() (in module jaxopt.linear_solve) solve_cg() (in module jaxopt.linear_solve) solve_cholesky() (in module jaxopt.linear_solve) solve_gmres() (in module jaxopt.linear_solve) solve_lu() (in module jaxopt.linear_solve) solve_normal_cg() (in module jaxopt.linear_solve) solver (jaxopt.AndersonWrapper attribute) (jaxopt.CvxpyQP attribute) (jaxopt.LevenbergMarquardt attribute)

solver() (jaxopt.LevenbergMarquardt method) sparse_plus() (in module jaxopt.loss) sparse_sigmoid() (in module jaxopt.loss) stepsize (jaxopt.BFGS attribute) (jaxopt.Broyden attribute) (jaxopt.GradientDescent attribute) (jaxopt.LBFGS attribute) (jaxopt.LBFGSB attribute) (jaxopt.MirrorDescent attribute) (jaxopt.ProjectedGradient attribute) (jaxopt.ProximalGradient attribute) stepsize_updates_frequency (jaxopt.BoxOSQP attribute) (jaxopt.OSQP attribute) stop_criterion (jaxopt.LevenbergMarquardt attribute) stop_if_linesearch_fails (jaxopt.Broyden attribute) (jaxopt.LBFGS attribute) (jaxopt.LBFGSB attribute)

T

termination_check_frequency (jaxopt.BoxOSQP attribute) (jaxopt.OSQP attribute) tol (jaxopt.AndersonAcceleration attribute) (jaxopt.ArmijoSGD attribute) (jaxopt.BacktrackingLineSearch attribute) (jaxopt.BFGS attribute) (jaxopt.Bisection attribute) (jaxopt.BlockCoordinateDescent attribute) (jaxopt.BoxCDQP attribute) (jaxopt.BoxOSQP attribute) (jaxopt.Broyden attribute) (jaxopt.EqualityConstrainedQP attribute) (jaxopt.FixedPointIteration attribute) (jaxopt.GaussNewton attribute) (jaxopt.GradientDescent attribute) (jaxopt.HagerZhangLineSearch attribute) (jaxopt.IterativeRefinement attribute) (jaxopt.LBFGS attribute) (jaxopt.LBFGSB attribute) (jaxopt.LevenbergMarquardt attribute) (jaxopt.MirrorDescent attribute) (jaxopt.NonlinearCG attribute) (jaxopt.OptaxSolver attribute) (jaxopt.OSQP attribute) (jaxopt.PolyakSGD attribute) (jaxopt.ProjectedGradient attribute) (jaxopt.ProximalGradient attribute) (jaxopt.ScipyBoundedMinimize attribute) (jaxopt.ScipyMinimize attribute) (jaxopt.ScipyRootFinding attribute)

tree_add() (in module jaxopt.tree_util) tree_add_scalar_mul() (in module jaxopt.tree_util) tree_div() (in module jaxopt.tree_util) tree_l2_norm() (in module jaxopt.tree_util) tree_mul() (in module jaxopt.tree_util) tree_scalar_mul() (in module jaxopt.tree_util) tree_sub() (in module jaxopt.tree_util) tree_sum() (in module jaxopt.tree_util) tree_vdot() (in module jaxopt.tree_util) tree_zeros_like() (in module jaxopt.tree_util)

U

unroll (jaxopt.AndersonAcceleration attribute) (jaxopt.AndersonWrapper attribute) (jaxopt.ArmijoSGD attribute) (jaxopt.BacktrackingLineSearch attribute) (jaxopt.BFGS attribute) (jaxopt.Bisection attribute) (jaxopt.BlockCoordinateDescent attribute) (jaxopt.BoxCDQP attribute) (jaxopt.BoxOSQP attribute) (jaxopt.Broyden attribute) (jaxopt.FixedPointIteration attribute) (jaxopt.GaussNewton attribute) (jaxopt.HagerZhangLineSearch attribute) (jaxopt.IterativeRefinement attribute) (jaxopt.LBFGS attribute) (jaxopt.LBFGSB attribute) (jaxopt.LevenbergMarquardt attribute) (jaxopt.MirrorDescent attribute) (jaxopt.NonlinearCG attribute) (jaxopt.OptaxSolver attribute) (jaxopt.PolyakSGD attribute) (jaxopt.ProjectedGradient attribute) (jaxopt.ProximalGradient attribute) update() (jaxopt.AndersonAcceleration method) (jaxopt.AndersonWrapper method) (jaxopt.ArmijoSGD method) (jaxopt.BacktrackingLineSearch method) (jaxopt.BFGS method) (jaxopt.Bisection method) (jaxopt.BlockCoordinateDescent method) (jaxopt.BoxCDQP method) (jaxopt.BoxOSQP method) (jaxopt.Broyden method) (jaxopt.FixedPointIteration method) (jaxopt.GaussNewton method) (jaxopt.GradientDescent method) (jaxopt.HagerZhangLineSearch method) (jaxopt.LBFGS method) (jaxopt.LBFGSB method) (jaxopt.LevenbergMarquardt method) (jaxopt.MirrorDescent method) (jaxopt.NonlinearCG method) (jaxopt.OptaxSolver method) (jaxopt.PolyakSGD method) (jaxopt.ProjectedGradient method) (jaxopt.ProximalGradient method)

update_state_using_delta_params() (jaxopt.LevenbergMarquardt method) update_state_using_gain_ratio() (jaxopt.LevenbergMarquardt method) upper (jaxopt.Bisection attribute) use_gamma (jaxopt.LBFGS attribute) (jaxopt.LBFGSB attribute) use_jacrev (jaxopt.ScipyRootFinding attribute)

V

value_and_grad (jaxopt.ArmijoSGD attribute) (jaxopt.BacktrackingLineSearch attribute) (jaxopt.BFGS attribute) (jaxopt.GradientDescent attribute) (jaxopt.HagerZhangLineSearch attribute) (jaxopt.LBFGS attribute) (jaxopt.LBFGSB attribute) (jaxopt.NonlinearCG attribute) (jaxopt.OptaxSolver attribute) (jaxopt.ProjectedGradient attribute) (jaxopt.ProximalGradient attribute) (jaxopt.ScipyMinimize attribute) verbose (jaxopt.AndersonAcceleration attribute) (jaxopt.AndersonWrapper attribute) (jaxopt.ArmijoSGD attribute) (jaxopt.BacktrackingLineSearch attribute) (jaxopt.BFGS attribute) (jaxopt.Bisection attribute) (jaxopt.BlockCoordinateDescent attribute) (jaxopt.BoxCDQP attribute) (jaxopt.BoxOSQP attribute) (jaxopt.Broyden attribute) (jaxopt.FixedPointIteration attribute) (jaxopt.GaussNewton attribute) (jaxopt.GradientDescent attribute) (jaxopt.HagerZhangLineSearch attribute) (jaxopt.IterativeRefinement attribute) (jaxopt.LBFGS attribute) (jaxopt.LBFGSB attribute) (jaxopt.LevenbergMarquardt attribute) (jaxopt.MirrorDescent attribute) (jaxopt.NonlinearCG attribute) (jaxopt.OptaxSolver attribute) (jaxopt.PolyakSGD attribute) (jaxopt.ProjectedGradient attribute) (jaxopt.ProximalGradient attribute)

X

xtol (jaxopt.LevenbergMarquardt attribute)