torch_sim.modelsΒΆ

Models integrated with TorchSim.

Modules

fairchem

FairChem model wrapper for torch-sim.

fairchem_legacy

Wrapper for Legacy FairChem ecosystem models in TorchSim.

graphpes

An interface for using arbitrary GraphPESModels in ts.

interface

Core interfaces for all models in TorchSim.

lennard_jones

Classical pairwise interatomic potential model.

mace

Wrapper for MACE model in TorchSim.

mattersim

TorchSim wrapper for MatterSim models.

metatomic

morse

Anharmonic interatomic potential for molecular dynamics.

nequip_framework

Wrapper for NequIP-Allegro models in TorchSim.

orb

ORB: PyTorch implementation of ORB models for atomistic simulations.

particle_life

Particle life model for computing forces between particles.

sevennet

TorchSim wrapper for SevenNet models.

soft_sphere

Soft sphere model for computing energies, forces and stresses.