xnn.common.featurizers.radial

Radial basis expansion shared across families (SchNet + equivariant GNN edge embedding).

Classes

GaussianRBF([n_rbf, cutoff, gamma])

Gaussian radial basis expansion (SchNet-style).