xnn.gnn.featurizers.cutoff

Polynomial cutoff envelope used by the equivariant GNN edge embedding (NequIP/MACE).

Classes

PolynomialCutoff(cutoff[, p])

Smooth p-degree polynomial envelope (used by NequIP/MACE), C^2 at rc.