Last released Nov 12, 2026
Certified first-order proximal point methods for composite optimization in JAX and Optax.