Implicit-Function Module

iskra.adjoint

compute_jacobian(fn[, in_idx, out_idx])
compute_jacobians(fn[, iterate_in, ...])
compute_numerical_jacobian(solver_fn[, ...])
make_adjoint_layer(solver_fn, implicit_fn[, ...])	Makes a solver differentiable via the adjoint method using a corresponding implicit function.
make_adjoint_vjps(fn[, param_args, ...])	Constructs the VJP functions needed for adjoint computation.
make_fixed_point_layer(fn[, iterates, ...])