Model wrappers
==============

These classes are used to wrap existing models and provide additional
functionalities, such as new outputs or modified behavior.

.. autoclass:: metatomic.torch.heat_flux.HeatFlux
    :members:
    :exclude-members: forward


.. autoclass:: metatomic.torch.dftd3.DFTD3
    :members:
    :exclude-members: forward