🧪 Experimental
===============

Experimental features subject to changes before being graduated into `optax`.

.. currentmodule:: optax.experimental