← all repositories

torchmd/torchmd-net

Neural network potential models (TensorNet, Transformers, GNN) for molecular dynamics simulation implemented in PyTorch.

476 stars Python Domain AppsML Frameworks
torchmd-net
Velocity · 7d
+0.3
★ / day
Trend
steady
star history

TorchMD-NET provides neural network architectures for predicting molecular potential energies and forces, which are critical for molecular dynamics simulations. It implements several equivariant neural network architectures including TensorNet, Equivariant Transformers, and Graph Neural Networks, all exposed as PyTorch modules. The trained potentials integrate with GPU-accelerated molecular dynamics codes like ACEMD, OpenMM, and TorchMD for running simulations.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.