cuequivariance#

Segmented Polynomials#

Operation(buffers)

Descriptor mapping input/output buffers to tensor product operands.

SegmentedOperand([segments, ndim, _dims])

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

SegmentedTensorProduct([...])

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

SegmentedPolynomial(inputs, outputs, operations)

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

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.

EquivariantPolynomial(inputs, outputs, ...)

A polynomial representation with equivariance constraints.

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.