NequIPLightningModule
#####################

.. autoclass:: nequip.train.NequIPLightningModule
    :members:

.. autoclass:: nequip.train.EMALightningModule
    :members:

.. autoclass:: nequip.train.ConFIGLightningModule
    :members:

.. autoclass:: nequip.train.EMAConFIGLightningModule
    :members:

.. autoclass:: nequip.train.ScheduleFreeLightningModule
    :members: