API reference

Note

This is the documentation for the atomistic capabilities of metatensor-torch version 0.5.2. For other versions, look in the following pages:

The atomistic machine learning models capabilities of metatensor are part of the TorchScript API, and are installed when installing the TorchScript API. We provide code for using atomistic systems as input of a machine learning model with metatensor.torch.atomistic.System, and exporting trained models with metatensor.torch.atomistic.MetatensorAtomisticModel.

C++ API reference

The C++ API can be accessed by including <metatensor/torch/atomistic.hpp>.