MolecularDiffusion.modules.models.ligandiff.layer_norm

GVP layer norm. Verbatim port of LigandDiff’s src/layer_norm.py.

Attributes

s_V

Classes

GVPLayerNorm

LayerNorm on the scalar channel, RMS rescaling on the vector channel.

Module Contents

class MolecularDiffusion.modules.models.ligandiff.layer_norm.GVPLayerNorm(dims: Tuple[int, int], eps: float = 1e-05)

Bases: torch.nn.Module

LayerNorm on the scalar channel, RMS rescaling on the vector channel.

forward(x: torch.Tensor | s_V) → torch.Tensor | s_V

eps

feature_layer_norm

MolecularDiffusion.modules.models.ligandiff.layer_norm.s_V