TorchScript backendΒΆ
We provide a PyTorch C++ extension defining all of the core metatomic types in a way compatible with the TorchScript compiler, allowing users to save and load models based on metatomic everywhere TorchScript is supported. This means one can define, train and save a model from Python, and then load it with pure C++ code, without requiring a Python interpreter. Please refer to the installation instructions to know how to install the Python and C++ sides of this library.
TorchScript Python API referenceRead the documentation for all the classes and functions in the
metatomic-torch Python package.
TorchScript C++ API referenceRead the documentation for all the classes and functions in the
metatomic/torch.hpp C++ header.