enerzyme.models.layers.gradient.EnergyVarianceGradientLayer# class enerzyme.models.layers.gradient.EnergyVarianceGradientLayer[source]# Bases: BaseFFLayer __init__() → None[source]# get_E_var_grad(E_var: Tensor, Ra: Tensor) → Tensor[source]#