Performance

Review performance benchmarks, memory requirements, and scaling characteristics for NIM for BGR.

Benchmark Metric

The primary benchmark metric is the average time per atom per optimization step:

• P: Time per atom per optimization step
• T: Total time to perform optimization of all structures in the dataset
• N: Number of structures
• A_i: Number of atoms in structure i
• S_i: Number of steps required to optimize structure i

This metric normalizes performance across workloads where both the atom count and the number of optimization steps vary by structure.

Batch Size Estimation

At startup, NIM for BGR automatically estimates the optimal batch size for each GPU by running benchmark calculations with representative structures.

INFO     | BGR:cuda:0 |  Atom count 40448 took 396.18 ms, 9.79 μs/atom (2 runs)
INFO     | BGR:cuda:0 |  Atom count 39516 took 391.91 ms, 9.92 μs/atom (2 runs)
INFO     | BGR:cuda:0 |  Atom count 41360 took 418.03 ms, 10.11 μs/atom (2 runs)
INFO     | BGR:cuda:0 |  Estimated batch size: 36876, max: 47136

Performance Results

The following benchmarks use the MACE-MPA-0 model and the OMat24 dataset.

Model Characteristics

Different machine learning interatomic potential (MLIP) models exhibit varying computational characteristics. MACE is the default, bundled model. TensorNet and AIMNet2 generally offer faster inference and lower memory usage.

Scaling Considerations

The following factors influence the overall throughput and resource utilization of NIM for BGR:

• Multi-GPU scaling: NIM for BGR automatically distributes work across all available GPUs, with each GPU maintaining its own batch queue.
• Structure size: Larger structures require more memory per structure, which reduces the effective batch size.
• Cell optimization: Enabling cell optimization adds stress tensor calculations and slightly increases computational cost.
• DFT-D3 corrections: Dispersion corrections add modest overhead to each evaluation.

Benchmark Environment

The following parameters define the benchmark reference environment: