Index

A

adjoint() (jaxfit.LinearOperator method), [1] arctan() (jaxfit.loss_functions.LossFunctionsJIT method) (jaxfit.LossFunctionsJIT method), [1] args (jaxfit.LinearOperator attribute), [1]

aslinearoperator() (in module jaxfit), [1] autdiff_jac() (jaxfit.least_squares.LeastSquares method) (jaxfit.LeastSquares method), [1] AutoDiffJacobian (class in jaxfit), [1] (class in jaxfit.least_squares)

B

build_quadratic_1d() (jaxfit.common_jax.CommonJIT method) (jaxfit.CommonJIT method), [1]

C

cauchy() (jaxfit.loss_functions.LossFunctionsJIT method) (jaxfit.LossFunctionsJIT method), [1] check_termination() (in module jaxfit), [1] (in module jaxfit.common_scipy) check_tolerance() (in module jaxfit), [1] (in module jaxfit.least_squares) check_x_scale() (in module jaxfit), [1] (in module jaxfit.least_squares) cho_factor() (in module jaxfit), [1] cho_solve() (in module jaxfit), [1] CL_scaling_vector() (in module jaxfit), [1] (in module jaxfit.common_scipy) CommonJIT (class in jaxfit), [1] (class in jaxfit.common_jax) compute_jac_scale() (jaxfit.common_jax.CommonJIT method) (jaxfit.CommonJIT method), [1] construct_all_loss_functions() (jaxfit.loss_functions.LossFunctionsJIT method) (jaxfit.LossFunctionsJIT method), [1] construct_single_loss_function() (jaxfit.loss_functions.LossFunctionsJIT method) (jaxfit.LossFunctionsJIT method), [1] copysign() (in module jaxfit), [1] create_ad_jacobian() (jaxfit.AutoDiffJacobian method), [1] (jaxfit.least_squares.AutoDiffJacobian method) create_arctan_funcs() (jaxfit.loss_functions.LossFunctionsJIT method) (jaxfit.LossFunctionsJIT method), [1] create_calculate_cost() (jaxfit.loss_functions.LossFunctionsJIT method) (jaxfit.LossFunctionsJIT method), [1] (jaxfit.trf.TrustRegionJITFunctions method) (jaxfit.TrustRegionJITFunctions method), [1] create_cauchy_funcs() (jaxfit.loss_functions.LossFunctionsJIT method) (jaxfit.LossFunctionsJIT method), [1] create_check_isfinite() (jaxfit.trf.TrustRegionJITFunctions method) (jaxfit.TrustRegionJITFunctions method), [1] create_covariance_svd() (jaxfit.CurveFit method), [1] (jaxfit.minpack.CurveFit method)

create_default_loss_func() (jaxfit.trf.TrustRegionJITFunctions method) (jaxfit.TrustRegionJITFunctions method), [1] create_grad_func() (jaxfit.trf.TrustRegionJITFunctions method) (jaxfit.TrustRegionJITFunctions method), [1] create_grad_hat() (jaxfit.trf.TrustRegionJITFunctions method) (jaxfit.TrustRegionJITFunctions method), [1] create_huber_funcs() (jaxfit.loss_functions.LossFunctionsJIT method) (jaxfit.LossFunctionsJIT method), [1] create_jac_sum() (jaxfit.common_jax.CommonJIT method) (jaxfit.CommonJIT method), [1] create_js_dot() (jaxfit.common_jax.CommonJIT method) (jaxfit.CommonJIT method), [1] create_quadratic_funcs() (jaxfit.common_jax.CommonJIT method) (jaxfit.CommonJIT method), [1] create_scale_for_robust_loss_function() (jaxfit.common_jax.CommonJIT method) (jaxfit.CommonJIT method), [1] create_scale_rhos() (jaxfit.loss_functions.LossFunctionsJIT method) (jaxfit.LossFunctionsJIT method), [1] create_sigma_transform_funcs() (jaxfit.CurveFit method), [1] (jaxfit.minpack.CurveFit method) create_soft_l1_funcs() (jaxfit.loss_functions.LossFunctionsJIT method) (jaxfit.LossFunctionsJIT method), [1] create_stack_rhos() (jaxfit.loss_functions.LossFunctionsJIT method) (jaxfit.LossFunctionsJIT method), [1] create_svd_funcs() (jaxfit.trf.TrustRegionJITFunctions method) (jaxfit.TrustRegionJITFunctions method), [1] create_zscale() (jaxfit.loss_functions.LossFunctionsJIT method) (jaxfit.LossFunctionsJIT method), [1] curve_fit() (in module jaxfit), [1] (in module jaxfit.minpack) (jaxfit.CurveFit method), [1] (jaxfit.minpack.CurveFit method) CurveFit (class in jaxfit), [1] (class in jaxfit.minpack)

D

dot() (jaxfit.LinearOperator method), [1]

E

evaluate_quadratic() (in module jaxfit), [1] (in module jaxfit.common_scipy) (jaxfit.common_jax.CommonJIT method) (jaxfit.CommonJIT method), [1]

F

find_active_constraints() (in module jaxfit), [1] (in module jaxfit.common_scipy)

G

get_empty_rhos() (jaxfit.loss_functions.LossFunctionsJIT method) (jaxfit.LossFunctionsJIT method), [1]

get_loss_function() (jaxfit.loss_functions.LossFunctionsJIT method) (jaxfit.LossFunctionsJIT method), [1]

H

H (jaxfit.LinearOperator property), [1] hess_inv (jaxfit.OptimizeResult attribute), [1]

huber() (jaxfit.loss_functions.LossFunctionsJIT method) (jaxfit.LossFunctionsJIT method), [1]

I

in_bounds() (in module jaxfit), [1] (in module jaxfit.common_scipy)

intersect_trust_region() (in module jaxfit), [1] (in module jaxfit.common_scipy) issparse() (in module jaxfit), [1]

J

jacfwd() (in module jaxfit), [1] jax_solve_triangular() (in module jaxfit), [1] jax_svd() (in module jaxfit), [1] jaxfit module, [1] jaxfit.common_jax module jaxfit.common_scipy module

jaxfit.least_squares module jaxfit.loss_functions module jaxfit.minpack module jaxfit.trf module jit() (in module jaxfit), [1]

L

least_squares() (jaxfit.least_squares.LeastSquares method) (jaxfit.LeastSquares method), [1] LeastSquares (class in jaxfit), [1] (class in jaxfit.least_squares)

LinAlgError, [1] LinearOperator (class in jaxfit), [1] LossFunctionsJIT (class in jaxfit), [1] (class in jaxfit.loss_functions)

M

make_strictly_feasible() (in module jaxfit), [1] (in module jaxfit.common_scipy) matmat() (jaxfit.LinearOperator method), [1] matvec() (jaxfit.LinearOperator method), [1] maxcv (jaxfit.OptimizeResult attribute), [1] message (jaxfit.OptimizeResult attribute), [1] minimize_quadratic_1d() (in module jaxfit), [1] (in module jaxfit.common_scipy) module jaxfit, [1] jaxfit.common_jax jaxfit.common_scipy jaxfit.least_squares jaxfit.loss_functions jaxfit.minpack jaxfit.trf setup tests

N

ndim (jaxfit.LinearOperator attribute), [1], [2], [3]

nit (jaxfit.OptimizeResult attribute), [1] norm() (in module jaxfit), [1]

O

OptimizeResult (class in jaxfit), [1]

OptimizeWarning, [1]

P

pad_fit_data() (jaxfit.CurveFit method), [1] (jaxfit.minpack.CurveFit method) prepare_bounds() (in module jaxfit), [1] (in module jaxfit.least_squares) print_header_linear() (in module jaxfit), [1] (in module jaxfit.common_scipy)

print_header_nonlinear() (in module jaxfit), [1] (in module jaxfit.common_scipy) print_iteration_linear() (in module jaxfit), [1] (in module jaxfit.common_scipy) print_iteration_nonlinear() (in module jaxfit), [1] (in module jaxfit.common_scipy)

R

reflective_transformation() (in module jaxfit), [1] (in module jaxfit.common_scipy)

rmatmat() (jaxfit.LinearOperator method), [1] rmatvec() (jaxfit.LinearOperator method), [1]

S

select_step() (jaxfit.trf.TrustRegionReflective method) (jaxfit.TrustRegionReflective method), [1] setup module soft_l1() (jaxfit.loss_functions.LossFunctionsJIT method) (jaxfit.LossFunctionsJIT method), [1] solve_lsq_trust_region() (in module jaxfit), [1] (in module jaxfit.common_scipy)

solve_trust_region_2d() (in module jaxfit), [1] (in module jaxfit.common_scipy) status (jaxfit.OptimizeResult attribute), [1] step_size_to_bound() (in module jaxfit), [1] (in module jaxfit.common_scipy) success (jaxfit.OptimizeResult attribute), [1]

T

T (jaxfit.LinearOperator property), [1] tests module transpose() (jaxfit.LinearOperator method), [1] tree_flatten() (in module jaxfit), [1] trf() (jaxfit.trf.TrustRegionReflective method) (jaxfit.TrustRegionReflective method), [1] trf_bounds() (jaxfit.trf.TrustRegionReflective method) (jaxfit.TrustRegionReflective method), [1]

trf_no_bounds() (jaxfit.trf.TrustRegionReflective method) (jaxfit.TrustRegionReflective method), [1] trf_no_bounds_timed() (jaxfit.trf.TrustRegionReflective method) (jaxfit.TrustRegionReflective method), [1] TrustRegionJITFunctions (class in jaxfit), [1] (class in jaxfit.trf) TrustRegionReflective (class in jaxfit), [1] (class in jaxfit.trf)

U

update_flength() (jaxfit.CurveFit method), [1] (jaxfit.minpack.CurveFit method) update_function() (jaxfit.least_squares.LeastSquares method) (jaxfit.LeastSquares method), [1]

update_tr_radius() (in module jaxfit), [1] (in module jaxfit.common_scipy)

W

warn() (in module jaxfit), [1]

wrap_jac() (jaxfit.least_squares.LeastSquares method) (jaxfit.LeastSquares method), [1]