API reference

This API reference contains a collection of all functions, modules, and objects available in dipm, along with detailed descriptions of their purpose and operation.

• Data processing
  • Graph Dataset Builder
  • Chemical System Datasets
  • Chemical System
  • Dataset Info
  • Dataset Configs
  • Atomic Number Table
  • Graph Definition
  • Graph Dataset
  • Data parallelization and batch prefetching
  • Data split
  • Helper functions
• Models
  • Base class
  • MACE
  • NequIP
  • ViSNet
  • LiTEN
  • EquiformerV2
  • UMA
  • Force Field predictor
  • Prediction
• Layers
  • Commonly used layers
  • ESCN Layers
• Model training
  • Training Loop
  • Training step
  • Optimizer
  • IO handling during training
  • Model evaluation
• Loss
  • Loss
• Simulations
  • Simulation Engine: Interface
  • Simulation Config
  • Simulation State
  • Simulation Enums
  • Create graph from ASE atoms
  • JAX-MD Simulation Engine
  • ASE Simulation Engine
  • MLIP ASE Calculator
  • Temperature scheduling
  • Batched inference
• Utils
  • Model I/O