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
istepandmd_timeparameters. - Electric Field Support: External electric field application for AIMNet2 models.
- Flexible Barostat: Isotropic and anisotropic cell shape changes for NPT.
- Configurable Output: Adjustable
save_intervalfor trajectory snapshot frequency.
Limitations
- The MACE-MPA-0 model automatically downloads using an NGC API key. You must provide AIMNet2 and TensorNet models externally using a volume mount.
- Periodic boundary conditions (PBC) mode is fixed at container startup. You cannot mix periodic and non-periodic structures in the same container instance.
- Each request processes a single structure. For high-throughput screening of many structures, use the Batched Geometry Relaxation (BGR) NIM.
- Very large structures can exceed GPU memory limits. Use the
/v1/statusendpoint to checkmax_system_sizefor your configuration. - Electric field (
efield) is only supported for AIMNet2 models.