API reference

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>.