Interactions#

Examples demonstrating GPU-accelerated computation of atomic interactions, including DFT-D3 dispersion corrections with environment-dependent C6 coefficients.

These examples show how to:

  • Compute dispersion energies and forces for molecules

  • Process batches of crystal structures

  • Integrate with PyTorch for differentiable workflows

DFT-D3 Dispersion Correction for a Molecule

DFT-D3 Dispersion Correction for a Molecule

DFT-D3 Dispersion Correction for Batched Crystals

DFT-D3 Dispersion Correction for Batched Crystals

Electrostatics#

Examples demonstrating GPU-accelerated computation of long-range electrostatic interactions in periodic systems using Coulomb, Ewald summation, and Particle Mesh Ewald (PME).

These examples show how to:

  • Compute direct Coulomb interactions (damped and undamped)

  • Use Ewald summation for periodic systems with automatic parameter estimation

  • Apply Particle Mesh Ewald (PME) for O(N log N) scaling

  • Work with neighbor list and neighbor matrix formats

  • Perform batch evaluation for multiple systems

  • Leverage autograd for computing forces and gradients

Coulomb Electrostatic Interactions

Coulomb Electrostatic Interactions

Ewald Summation for Long-Range Electrostatics

Ewald Summation for Long-Range Electrostatics

Particle Mesh Ewald (PME) for Long-Range Electrostatics

Particle Mesh Ewald (PME) for Long-Range Electrostatics

Neighbor Lists#

Examples demonstrating efficient O(N) neighbor list construction using GPU-accelerated cell list algorithms.

These examples show how to:

  • Build neighbor lists for single and batched systems

  • Use dense or sparse COO output formats

  • Detect when neighbor lists need rebuilding

  • Optimize performance with torch.compile

  • Integrate with molecular dynamics workflows

Simple Neighbor List Example

Simple Neighbor List Example

Batch Neighbor List Example

Batch Neighbor List Example

Neighbor List Rebuild Detection Example

Neighbor List Rebuild Detection Example

Torch.compile Performance Benefits Example

Torch.compile Performance Benefits Example

Gallery generated by Sphinx-Gallery