xnn.common.models#
Modules
The contract every model obeys. |
|
Global charge solve for LES: charge equilibration with the latent Ewald kernel. |
|
Grimme DFT-D3 London dispersion: a geometry-dependent correction for any xnn model. |
|
DFT-D4 London dispersion: a charge-dependent correction for any xnn model. |
|
Shared machinery of the dispersion add-ons (DFT-D3, DFT-D4). |
|
Fast paths of the dispersion terms (DFT-D3, DFT-D4). |
|
Large-system EEQ charges: matrix-free operator, solvers, implicit differentiation. |
|
Point-charge electrostatics shared by the charge-predicting models. |
|
Fast paths: optimized kernels behind the same parameters as the reference code. |
|
Model hub: one-line |
|
Latent Ewald Summation (LES): long-range interactions for any xnn model. |
|
Tensor operations shared across model families (gnn / cnn / dnn). |
|
Differentiate energy to get conservative forces and the stress tensor. |
|
Smooth particle-mesh Ewald reciprocal sums (Essmann and co-workers, 1995). |
|
Reciprocal-space sums through factorized phases (the Ewald fast paths). |
|
Recomputation blocks whose memory stays bounded through a double backward. |
|
Name -> class registry so new models are added without touching the core. |