cuequivariance#

Segmented Polynomials#

Operation

Descriptor mapping input/output buffers to tensor product operands.

SegmentedOperand

A segmented operand is a list of segment's shapes.

SegmentedTensorProduct

Irreps-agnostic and dataflow-agnostic descriptor of a segmented tensor product

SegmentedPolynomial

A polynomial representation using segmented tensor products.

Group Representations#

Rep

Abstract Class, Representation of a Lie group.

Irrep

Subclass of Rep for an irreducible representation of a Lie group.

SO3

Subclass of Irrep, real irreducible representations of the 3D rotation group \(SO(3)\).

O3

Subclass of Irrep, real irreducible representations of the 3D rotation group \(O(3)\).

SU2

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 and polynomials.

Irreps

Direct sum of irreducible representations with multiplicities.

IrrepsLayout

Enum for the possible data layouts.

IrrepsAndLayout

A group representation (Rep) made from the combination of Irreps and IrrepsLayout into a single object.

EquivariantPolynomial

A polynomial representation with equivariance constraints.

Descriptors#

List of Descriptors

Utilities#

assume

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