TorchANI (J. Chem. Inf. Model. 2020)

github.com/aiqm/torchani
Active563updated 1 month ago
Python
MIT

Official free and open-source PyTorch implementation of the ANI family of neural network interatomic potentials, shipping pre-trained ANI-1ccx/ANI-1x/ANI-2x ensembles with energy and force training, built-in unit conversion, and ASE integration; the foundational ANI ecosystem widely used for organic and biomolecular ML force field workflows (AIQMS, 560+ stars, MIT License, actively maintained)

Sourced from

  • Awesome AI for Science — github.com/aiqm/torchani
  • GitHub — github.com/aiqm/torchani

Related resources

Neural differential equations in PyTorch

Stale1.6K2 years ago
Jupyter Notebook
Apache-2.0

E(3)-equivariant neural network interatomic potentials achieving DFT accuracy with up to 1000× less training data than invariant models, foundational architecture behind MACE and Allegro (Harvard, MIT, Nature Communications 2022)

Active9691 week ago
Python
MIT

PyTorch toolkit for deep neural networks in atomistic simulations, implementing SchNet, DimeNet++, PaiNN, and GemNet for molecular dynamics and quantum chemistry (900+ stars)

Active9412 weeks ago
Python
NOASSERTION

Highly scalable equivariant deep learning interatomic potentials enabling million-atom molecular dynamics simulations with ab initio accuracy, building on E(3)-equivariant architectures for large-scale atomistic modeling (mir-group, MIT License, 480+ stars)

Active4984 months ago
Python
MIT

Neural network-based exchange-correlation functional for density functional theory (DFT) that surpasses state-of-the-art hybrid functionals in accuracy for main-group thermochemistry, kinetics, and non-covalent interactions at semi-local DFT cost; includes PySCF/GPU4PySCF/ASE bindings and C++/Fortran integrations (248+ stars, MIT License)

Active2666 days ago
Python
MIT

Pretrained machine-learned force field for (bio)molecular simulations combining the fast SO3krates neural network for semi-local interactions with universal pairwise force fields for short-range repulsion, long-range electrostatics, and dispersion interactions; supports geometry optimization, NVT/NPT/NVE MD, fine-tuning, ASE calculator, and JAX-MD integration (JACS 2025, 218+ stars, MIT License)

Active2315 months ago
Python
MIT