jaxsnn.event.root.custom_root_newton

Classes

partial(func, *args, **keywords) - new function with partial application of the given arguments and keywords.

Functions

jaxsnn.event.root.custom_root_newton.cr_newton_solver(f, initial_guess, state, dt)

jaxsnn.event.root.custom_root_newton.solve(f, initial_guess, dt)

Newton’s method for root-finding.

Does not need to be differentiable