linear

impose_linear_equivariance(lin, in_rep, out_rep)	Impose equivariance constraints on a given torch.nn.Linear layer using torch parametrizations.
eLinear(in_rep, out_rep[, bias, ...])	Parameterize a \(\mathbb{G}\)-equivariant linear map with optional invariant bias.
eAffine(in_rep[, bias, learnable, init_scheme])	Equivariant affine map with per-irrep scales and invariant bias.
InvariantBias(in_rep)	Module parameterizing a learnable \(\mathbb{G}\)-invariant bias.