Jax Custom JVP with Lmfit Minimization
I’m looking to pass a jax calculated jacobian to my objective function (for an lmfit minimizer) that incorporates a BSpline object which is not supported in jax.
Jax Custom JVP with Lmfit Minimization
I’m looking to pass a jax calculated jacobian to my objective function (for an lmfit minimizer) that incorporates a BSpline object which is not supported in jax.