JAX Dataclass Utility
=====================

.. automodule:: nnx_ppo.jax_dataclass
   :members:
   :special-members: __init__
   :show-inheritance: