API reference¶
Note
This is the documentation for the atomistic capabilities of
metatensor-torch
version 0.5.4. For other versions,
look in the following pages:
Version 0.5
0.5.4
0.5.3
0.5.2
0.5.1
0.5.0
Version 0.4
0.4.0
Version 0.3
0.3.0
Older versions
0.2.1
0.2.0
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>
.