jaxmodel.autodiff

jaxmodel.autodiff.build_grad_y_objective(obj_flat)[source]

jaxmodel.autodiff.build_hess_y_objective(obj_flat)[source]

jaxmodel.autodiff.build_diag_hess_y_objective(grad_fun)[source]

jaxmodel.autodiff.build_jac_y_vector_fun(vec_fun)[source]