Averaging predictions per atom
##############################

.. automodule:: metatrain.utils.per_atom
    :members:
    :undoc-members:
    :show-inheritance: