optimistix_bfgs_scalar_function_minimization_quickstart.py

1import jax.numpy as jnp
2import optimistix as optx
3
4def fn(x, args):
5    return jnp.sum((x - 1) ** 2)
6
7solver = optx.BFGS(rtol=1e-6, atol=1e-6)
8y0 = jnp.array([0.0, 0.0])
9sol = optx.minimize(fn, solver, y0)
10
11print(f"Minimum found at: {sol.value}")