compat
------------------------

.. automodule:: flax.nnx.compat
.. currentmodule:: flax.nnx.compat