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

NequixModel

Nequix model wrapper for torch-sim.