Skip to main content
Ctrl+K

ALCHEMI Toolkit

  • User Guide
  • Supported Models
  • Examples
  • API
  • Changelog
  • Github
  • User Guide
  • Supported Models
  • Examples
  • API
  • Changelog
  • Github

Section Navigation

Modules

  • Data module (AtomicData, Batch, readers/writers)
    • nvalchemi.data.AtomicData
    • nvalchemi.data.Batch
    • nvalchemi.data.AtomicDataZarrWriter
    • nvalchemi.data.AtomicDataZarrReader
    • nvalchemi.data.Dataset
    • nvalchemi.data.DataLoader
    • nvalchemi.data.Reader
    • nvalchemi.data.datapipes.ZarrArrayConfig
    • nvalchemi.data.datapipes.ZarrWriteConfig
  • Hooks — Core Framework
    • nvalchemi.hooks.Hook
    • nvalchemi.hooks.HookContext
    • nvalchemi.hooks.HookRegistryMixin
    • nvalchemi.hooks.BiasedPotentialHook
    • nvalchemi.hooks.NeighborListHook
    • nvalchemi.hooks.WrapPeriodicHook
  • Dynamics module
    • Architecture Overview
    • Dynamics Hooks — Stages & Usage
    • Convergence Criteria
    • FusedStage — Single-GPU Orchestration
    • DistributedPipeline — Multi-GPU Workflows
    • Buffers & Data Flow
    • Implementing Custom Dynamics
    • API Reference
      • nvalchemi.dynamics.BaseDynamics
      • nvalchemi.dynamics.DemoDynamics
      • nvalchemi.dynamics.FusedStage
      • nvalchemi.dynamics.DistributedPipeline
      • nvalchemi.dynamics.DynamicsStage
      • nvalchemi.dynamics.ConvergenceHook
      • nvalchemi.dynamics.hooks.ConvergedSnapshotHook
      • nvalchemi.dynamics.hooks.EnergyDriftMonitorHook
      • nvalchemi.dynamics.hooks.FreezeAtomsHook
      • nvalchemi.dynamics.hooks.LoggingHook
      • nvalchemi.dynamics.hooks.MaxForceClampHook
      • nvalchemi.dynamics.hooks.NaNDetectorHook
      • nvalchemi.dynamics.hooks.ProfilerHook
      • nvalchemi.dynamics.hooks.SnapshotHook
      • nvalchemi.dynamics.DataSink
      • nvalchemi.dynamics.GPUBuffer
      • nvalchemi.dynamics.HostMemory
      • nvalchemi.dynamics.ZarrData
      • nvalchemi.dynamics.SizeAwareSampler
  • Models module (BaseModelMixin, ModelConfig, wrappers)
    • nvalchemi.models.base.ModelConfig
    • nvalchemi.models.base.BaseModelMixin
    • nvalchemi.models.demo.DemoModel
    • nvalchemi.models.demo.DemoModelWrapper
    • nvalchemi.models.mace.MACEWrapper
    • nvalchemi.models.aimnet2.AIMNet2Wrapper
    • nvalchemi.models.lj.LennardJonesModelWrapper
    • nvalchemi.models.dftd3.DFTD3ModelWrapper
    • nvalchemi.models.pme.PMEModelWrapper
    • nvalchemi.models.ewald.EwaldModelWrapper
    • nvalchemi.models.pipeline.PipelineModelWrapper
    • nvalchemi.models.pipeline.PipelineStep
    • nvalchemi.models.pipeline.PipelineGroup
  • Typing module (type aliases, enums, protocols)
  • API Reference

API Reference#

This is the API reference for NVIDIA ALCHEMI Toolkit. Please refer to the full user guide and examples for further details on how to use this package as API documentation alone may not provide sufficient information.

Modules

  • Data module (AtomicData, Batch, readers/writers)
    • Core classes
    • I/O and pipelines
    • Write configuration
  • Hooks — Core Framework
    • The Hook protocol
    • HookContext
    • Registration and dispatch
    • Task-category specialization
    • General-purpose hooks
    • API Reference
  • Dynamics module
    • Architecture Overview
    • Dynamics Hooks — Stages & Usage
    • Convergence Criteria
    • FusedStage — Single-GPU Orchestration
    • DistributedPipeline — Multi-GPU Workflows
    • Buffers & Data Flow
    • Implementing Custom Dynamics
    • API Reference
  • Models module (BaseModelMixin, ModelConfig, wrappers)
    • Core classes
    • Demo utilities
    • Machine-learned potentials
    • Physical / classical models
    • Composition
  • Typing module (type aliases, enums, protocols)

previous

Monitoring a Distributed Pipeline: Per-Rank Logging and Profiling

next

Data module (AtomicData, Batch, readers/writers)

Show Source

© Copyright 2026, NVIDIA.

Created using Sphinx 9.1.0.

Built with the PyData Sphinx Theme 0.15.2.