jaxmat.solvers.custom_newton module

newton_solve_jittable(f, x0, args, solver, *, tol=1e-06, maxiter=20, damping=1.0, has_aux=False, jac='fwd')

Parameters:

• f (Callable)

• x0 (Any)

• args (Any)

• solver (AbstractLinearSolver)

• tol (float)

• maxiter (int)

• damping (float)

• has_aux (bool)