torch_sim.modelsΒΆ

Models integrated with TorchSim.

Modules

torch_sim.models.fairchem

Wrapper for FairChem ecosystem models in TorchSim.

torch_sim.models.graphpes

An interface for using arbitrary GraphPESModels in torch_sim.

torch_sim.models.interface

Core interfaces for all models in torchsim.

torch_sim.models.lennard_jones

Classical pairwise interatomic potential model.

torch_sim.models.mace

Wrapper for MACE model in TorchSim.

torch_sim.models.mattersim

TorchSim wrapper for MatterSim models.

torch_sim.models.morse

Anharmonic interatomic potential for molecular dynamics.

torch_sim.models.orb

ORB: PyTorch implementation of ORB models for atomistic simulations.

torch_sim.models.particle_life

Particle life model for computing forces between particles.

torch_sim.models.sevennet

TorchSim wrapper for SevenNet models.

torch_sim.models.soft_sphere

Soft sphere model for computing energies, forces and stresses.