nequip.train.callbacks
######################

.. autoclass:: nequip.train.callbacks.SoftAdapt
    :members:

.. autoclass:: nequip.train.callbacks.LossCoefficientScheduler
    :members:

.. autoclass:: nequip.train.callbacks.LinearLossCoefficientScheduler
    :members:

.. autoclass:: nequip.train.callbacks.LossCoefficientMonitor
    :members:

.. autoclass:: nequip.train.callbacks.TestTimeXYZFileWriter
    :members:

.. autoclass:: nequip.train.callbacks.ValTimeXYZFileWriter
    :members:

.. autoclass:: nequip.train.callbacks.WandbWatch
    :members:

.. autoclass:: nequip.train.callbacks.TF32Scheduler
    :members: