TorchSim: An efficient atomistic simulation engine in PyTorch

Orion Cohen, Janosh Riebesell, Rhys Goodall +6

We introduce TorchSim, an open-source atomistic simulation engine tailored for the Machine Learned Interatomic Potential (MLIP) era. By rewriting core atomistic simulation primitiv…