e3nn_mlx.o3.Irreps

class e3nn_mlx.o3.Irreps(spec: 'str | Irrep | Irreps | Iterable[MulIrrep | Irrep | str | tuple[int, Irrep | str]]' = ())

Parameters:

spec (str | Irrep | Irreps | Iterable[MulIrrep | Irrep | str | tuple[int, Irrep | str]])

__init__(spec=())

Parameters:

spec (str | Irrep | Irreps | Iterable[MulIrrep | Irrep | str | tuple[int, Irrep | str]])

Return type:

None

Methods

D_from_angles(alpha, beta, gamma[, k])	Block-diagonal Wigner-D matrix for YXY Euler angles (requires a runtime).
D_from_axis_angle(axis, angle)
D_from_matrix(matrix)	Block-diagonal Wigner-D from a 3x3 matrix, honouring improper rotations.
D_from_quaternion(quaternion[, k])
__init__([spec])
count(irrep)
extend(other)
filter([keep, drop, lmax])
index(part)	Return the position of part, like upstream's tuple-derived index.
randn(*shape[, normalization, dtype])	Random array whose -1 axis carries this representation.
regroup()
remove_zero_multiplicities()
simplify()
slices()
sort()
spherical_harmonics(lmax[, p])

Attributes

parts
dim
lmax
ls
num_irreps
slice_by_mul