cuequivariance

Group Representations

Rep()	Abstract Class, Representation of a Lie group.
Irrep()	Subclass of Rep for an irreducible representation of a Lie group.
SO3(l)	Subclass of Irrep, real irreducible representations of the 3D rotation group \(SO(3)\).
O3(l, p)	Subclass of Irrep, real irreducible representations of the 3D rotation group \(O(3)\).
SU2(j)	Subclass of Irrep, irreducible representations of the Lie group \(SU(2)\).

clebsch_gordan(rep1, rep2, rep3)

Compute the Clebsch-Gordan coefficients.

Equivariant Tensor Products

These classes represent tensor products.

Irreps(*args)	Direct sum of irreducible representations with multiplicities.
IrrepsLayout(*values)	Enum for the possible data layouts.
IrrepsAndLayout(irreps[, layout])	A group representation (Rep) made from the combination of Irreps and IrrepsLayout into a single object.
SegmentedTensorProduct(*[, operands, paths, ...])	Irreps-agnostic and dataflow-agnostic descriptor of a segmented tensor product
EquivariantTensorProduct(d, operands[, ...])	Descriptor of an equivariant tensor product.
Operation(buffers)	Descriptor mapping input/output buffers to tensor product operands.

Descriptors

List of Descriptors

Utilities

assume([irrep_class, layout])

assume is a context manager or decorator to assume the irrep class and layout for a block of code or a function.