torch_sim.models.nequip_framework¶
Wrapper for NequIP-Allegro models in TorchSim.
This module re-exports the NequIP framework’s torch-sim integration for convenient importing. The actual implementation is maintained in the NequIP package.
References
NequIP Package: https://github.com/mir-group/nequip
Classes
NequIP model framework wrapper for torch-sim. |