Overview#
The NVIDIA NIM for Batch Molecular Dynamics provides a high-performance engine for batched molecular dynamics (MD) simulations, enabling researchers to study the time evolution of atomic systems using machine learning interatomic potentials (MLIPs). Through supporting multiple MLIP models including MACE, AIMNet2, and TensorNet optimized for NVIDIA GPUs, this NIM enables researchers to perform MD simulations at scale with near-Density Functional Theory (DFT) accuracy at a fraction of the computational cost.
Molecular dynamics is fundamental to understanding material properties, chemical reactions, protein dynamics, and many other phenomena in computational chemistry and materials science.
The NIM is primarily recommended for the following use cases:
High-throughput MD simulations: Ideal for studying thermodynamic properties, phase transitions, and transport phenomena across many systems.
Materials science research: Investigate thermal conductivity, diffusion coefficients, melting points, and other temperature-dependent properties.
Drug discovery workflows: Study ligand binding, conformational sampling, and free energy calculations with GPU-accelerated MD.
Reaction dynamics: Explore reactive systems and transition states with accurate MLIP-driven dynamics.
Key Features of NIM for BMD#
The NIM for BMD includes the following key features:
Multiple MLIP Models: Supports various machine learning interatomic potentials including MACE, AIMNet2 (with NSE variant for organic molecular systems), and TensorNet models, allowing users to choose the best model for their specific application.
Dynamic Batching: Optimizes GPU utilization by dynamically batching concurrent MD requests based on available GPU memory, maximizing throughput and efficiency.
Multiple Ensembles: Supports NVE (microcanonical), NVT (canonical with Langevin thermostat), and NPT (isothermal-isobaric with Monte Carlo barostat) ensembles for diverse simulation needs.
Trajectory Output: Returns complete trajectory snapshots including positions, velocities, energies, and optionally stress tensors for post-processing and analysis.
Restart Capability: Simulations can be seamlessly restarted from any saved state, enabling long simulations to be run in chunks and facilitating checkpoint and restart workflows.
Electric Field Support: For AIMNet2 models, external electric fields can be applied to study field-dependent phenomena.
Flexible Configuration: Per-request simulation parameters allow fine-grained control over temperature, timestep, ensemble, and barostat settings.
Supported Ensembles#
The NIM supports the following simulation ensembles:
Ensemble |
Settings |
Description |
|---|---|---|
NVE |
|
Microcanonical - constant energy |
NVT |
|
Canonical - Langevin thermostat |
NPT |
|
Isothermal-isobaric - MC barostat |
Supported Models#
Pre-bundled model: MACE-MP-0b2-Large is pre-bundled and auto-downloaded when the container is run with an NGC API key. Inference can be started immediately without downloading or mounting any model files.
NIM for BMD supports multiple MLIP model architectures. The ALCHEMI_NIM_MODEL_TYPE environment variable selects the architecture; the specific model variant is determined by the files mounted into the container (or the bundled MACE-MP-0b2-Large when using the default).
The following table lists the supported models and their capabilities:
|
Supported Models |
Periodic Only |
Non-periodic |
Bundled |
Reference |
|---|---|---|---|---|---|
|
MACE-MP-0b2-Large (built-in); other MACE models (for example, MACE-MPA-0) can be mounted |
Yes |
No |
MACE-MP-0b2-Large auto-downloaded |
|
|
TensorNet-MatPES-PBE-v2025.1-PES, TensorNet-MatPES-r2SCAN-v2025.1-PES |
Yes |
No |
No |
|
|
All AIMNet2 models |
No |
Yes |
No |
TensorNet models are MatPES potentials from materialyzeai/matgl (PyG version, not DGL).
AIMNet2 supports all models listed at isayevlab/aimnetcentral.
For non-bundled models, see Model Configuration for mounting instructions. For download hints and container launch arguments by model type, see Custom Models.
References#
Advantages of NIMs#
NIMs offer a simple and easy-to-deploy route for self-hosted AI applications. Two major advantages that NIMs offer for system administrators and developers are:
Increased productivity — NIMs allow developers to build generative AI applications quickly, in minutes rather than weeks, by providing a standardized way to add AI capabilities to their applications.
Simplified deployment — NIMs provide containers that can be easily deployed on various platforms, including clouds, data centers, or workstations, making it convenient for developers to test, deploy, and scale their applications.
The NIM for BMD provides fast, accurate models behind a consistent API for molecular dynamics simulations. As part of the broader NVIDIA NIM ecosystem, NIM for BMD can be used in conjunction with other NIMs such as the BGR (Batched Geometry Relaxation) NIM to build comprehensive atomistic modeling pipelines.