squarem_jaxopt
