Skip to main content

PyTorch implementation of ANI

Project description

Accurate Neural Network Potential on PyTorch


PyPI PyPI - Downloads


Actions Status Actions Status Actions Status Actions Status Actions Status Actions Status CodeFactor Total alerts


Actions Status Actions Status

We only provide compatibility with nightly PyTorch, but you can check if stable PyTorch happens to be supported by looking at the following badge:

Actions Status

TorchANI is a pytorch implementation of ANI. It is currently under alpha release, which means, the API is not stable yet. If you find a bug of TorchANI, or have some feature request, feel free to open an issue on GitHub, or send us a pull request.


TorchANI requires the latest preview version of PyTorch. You can install PyTorch by the following commands (assuming cuda10):

pip install numpy
pip install --pre torch torchvision -f

If you updated TorchANI, you may also need to update PyTorch:

pip install --upgrade --pre torch torchvision -f

After installing the correct PyTorch, you can install TorchANI by:

pip install torchani

See also PyTorch's official site for instructions of installing latest preview version of PyTorch.

To run the tests and examples, you must manually download a data package



Please cite the following paper if you use TorchANI

  • Xiang Gao, Farhad Ramezanghorbani, Olexandr Isayev, Justin S. Smith, and Adrian E. Roitberg. TorchANI: A Free and Open Source PyTorch Based Deep Learning Implementation of the ANI Neural Network Potentials. Journal of Chemical Information and Modeling 2020 60 (7), 3408-3415, DOI for Citing

JCIM Cover

ANI model parameters

All the ANI model parameters including (ANI2x, ANI1x, and ANI1ccx) are accessible from the following repositories:


To install TorchANI from GitHub:

git clone
cd torchani
pip install -e .

After TorchANI has been installed, you can build the documents by running sphinx-build docs build. But make sure you install dependencies:

pip install sphinx sphinx-gallery pillow matplotlib sphinx_rtd_theme

To manually run unit tests, do python nosetests

If you opened a pull request, you could see your generated documents at after you docs check succeed. Keep in mind that this repository is only for the purpose of convenience of development, and only keeps the latest push. The CI runing for other pull requests might overwrite this repository. You could rerun the docs check to overwrite this repo to your build.

Note to TorchANI developers

Never commit to the master branch directly. If you need to change something, create a new branch, submit a PR on GitHub.

You must pass all the tests on GitHub before your PR can be merged.

Code review is required before merging pull request.

Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

torchani-2.2.tar.gz (14.3 MB view hashes)

Uploaded source

Built Distribution

torchani-2.2-py3-none-any.whl (10.8 MB view hashes)

Uploaded py3

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring Facebook / Instagram Facebook / Instagram PSF Sponsor Fastly Fastly CDN Google Google Object Storage and Download Analytics Huawei Huawei PSF Sponsor Microsoft Microsoft PSF Sponsor NVIDIA NVIDIA PSF Sponsor Pingdom Pingdom Monitoring Salesforce Salesforce PSF Sponsor Sentry Sentry Error logging StatusPage StatusPage Status page