torch_sim.models.nequix¶
Wrapper for Nequix models in TorchSim.
This module re-exports the nequix model’s torch-sim integration for convenient importing. The actual implementation is maintained in the nequix package.
References
nequix Package: https://github.com/atomicarchitects/nequix
Classes
Nequix model wrapper for torch-sim. |