API Reference

This is the API reference for NVIDIA ALCHEMI Toolkit-Ops. 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.

PyTorch Reference

• nvalchemiops.torch.neighbors: Neighbor Lists
  • High-Level Interface
  • Unbatched Algorithms
  • Batched Algorithms
  • Rebuild Detection
  • Utility Functions
• nvalchemiops.torch.interactions.dispersion: Dispersion Corrections
  • High-Level Interface
  • Data Structures
  • Internal Custom Operators
• nvalchemiops.torch.interactions.electrostatics: Electrostatics
  • High-Level Interface
  • Coulomb Interactions
  • DSF Coulomb
  • Ewald Components
  • PME Components
  • K-Vector Generation
  • Parameter Estimation
• nvalchemiops.torch: Dynamics Optimizers
  • PyTorch Adapters for nvalchemiops
  • FIRE2 Optimizer

JAX Reference

• nvalchemiops.jax.neighbors: Neighbor Lists
  • High-Level Interface
  • Unbatched Algorithms
  • Batched Algorithms
  • Rebuild Detection
  • Exceptions
  • Utility Functions
• nvalchemiops.jax.interactions.dispersion: Dispersion Corrections
  • High-Level Interface
  • Data Structures
  • Internal Implementation
• nvalchemiops.jax.interactions.electrostatics: Electrostatics
  • High-Level Interface
  • Coulomb Interactions
  • Ewald Components
  • PME Components
  • K-Vector Generation
  • Parameter Estimation

Warp Reference

• nvalchemiops.neighbors: Neighbor Lists
  • Warp-Level Interface
• nvalchemiops.interactions.dispersion: Dispersion Corrections
  • Warp-Level Interface
• nvalchemiops.interactions.electrostatics: Electrostatic Interactions (Warp)
  • Electrostatics Interactions Module
  • Core Warp Kernels
• nvalchemiops.dynamics: Molecular Dynamics
  • Dynamics Module
  • Warp-Level Interface
  • Integrators
  • Optimizers
  • Thermostat Utilities
  • Cell Utilities
  • Cell Filter Utilities (Variable-Cell Optimization)
  • Constraint Utilities (SHAKE/RATTLE)
• nvalchemiops.math and nvalchemiops.torch.types: Utilities
  • Mathematical Utilities
  • Type Conversion Utilities