squarem-JAXopt
==============

This page contains the API documentation for squarem-JAXopt.

SquaremAcceleration
-------------------

.. autoclass:: squarem_jaxopt.SquaremAcceleration
   :exclude-members: init_state, squarem_step, update, optimality_fun, __init__, __post_init__, run, fixed_point_fun, maxiter, tol, has_aux, verbose, implicit_diff, implicit_diff_solve, jit, unroll