Skip to navigation

Release 1.0.0

Summary

This is the first release of NVIDIA NIM for Batched Molecular Dynamics.

Features

  • Multiple MLIP Models: Support for MACE, AIMNet2 (including NSE variant), and TensorNet models.
  • Multiple Ensembles: NVE (microcanonical), NVT (Langevin thermostat), and NPT (Monte Carlo barostat) ensembles.
  • Trajectory Output: Returns complete trajectory with positions, velocities, energies, and optionally stress tensors.
  • Restart Capability: Seamless restart from any saved state using the istep and md_time parameters.
  • Electric Field Support: External electric field application for AIMNet2 models.
  • Flexible Barostat: Isotropic and anisotropic cell shape changes for NPT.
  • Configurable Output: Adjustable save_interval for trajectory snapshot frequency.