Skip to main content
Ctrl+K

ALCHEMI Toolkit

  • User Guide
  • Supported Models
  • Examples
  • Change Log
  • API
  • Github
  • User Guide
  • Supported Models
  • Examples
  • Change Log
  • API
  • 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
  • Dynamics module
    • Architecture Overview
    • Hooks — Observe & Modify
    • 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.Hook
      • nvalchemi.dynamics.HookStageEnum
      • nvalchemi.dynamics.ConvergenceHook
      • nvalchemi.dynamics.hooks.BiasedPotentialHook
      • 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.NeighborListHook
      • nvalchemi.dynamics.hooks.ProfilerHook
      • nvalchemi.dynamics.hooks.SnapshotHook
      • nvalchemi.dynamics.hooks.WrapPeriodicHook
      • 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.ModelCard
    • nvalchemi.models.base.BaseModelMixin
    • nvalchemi.models.demo.DemoModel
    • nvalchemi.models.demo.DemoModelWrapper
    • nvalchemi.models.mace.MACEWrapper
    • nvalchemi.models.lj.LennardJonesModelWrapper
    • nvalchemi.models.dftd3.DFTD3ModelWrapper
    • nvalchemi.models.pme.PMEModelWrapper
    • nvalchemi.models.ewald.EwaldModelWrapper
    • nvalchemi.models.composable.ComposableModelWrapper
  • 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
  • Dynamics module
    • Architecture Overview
    • Hooks — Observe & Modify
    • 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

Change Log

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.