API Reference#
Core classes#
Base class for all dynamics simulations. |
|
Velocity Verlet integrator for molecular dynamics simulations. |
|
Composite dynamics engine fusing multiple sub-stages on a single GPU. |
|
Orchestrates multi-rank pipeline execution. |
Protocols and enums#
Enumeration of stages in the dynamics step where hooks can fire. |
Convergence#
Hook that evaluates composable convergence criteria and optionally migrates converged samples between pipeline stages. |
Hooks#
Dynamics-specific hooks are documented in Dynamics Hooks — Stages & Usage.
General-purpose hooks (NeighborListHook,
BiasedPotentialHook,
WrapPeriodicHook), the shared profiling hooks
(StageTimingHook,
TorchProfilerHook), and the core hook protocol
are documented in Hooks — Core Framework.
Data sinks#
Abstract base class for local storage of Batch data. |
|
GPU-resident buffer for storing batched atomic data. |
|
CPU-resident buffer for storing batched atomic data. |
|
Zarr-backed storage for batched atomic data. |
Sampling#
Size-aware sampler for inflight batching. |