Install & Compatibility
Where this runs
No compatibility data collected yet for this library.
Code
Verified usage
Verified import paths — ran on the pinned version, not inferred.
oe
✓ from openequivariance import oe
✗ import openequivariance
Top-level module is 'oe', not package name. Use 'from openequivariance import oe'.
TPProblem
✓ from openequivariance import TPProblem
Minimal tensor product using Torch backend on CPU.
import torch
from openequivariance import oe
irreps_in1 = [(1, (1, 1)), (1, (1, 2)), (1, (2, 1)), (1, (2, 2))]
irreps_in2 = [(1, (1, 1)), (1, (1, 2)), (1, (2, 1)), (1, (2, 2))]
irreps_out = [(1, (1, 1)), (1, (1, 2)), (1, (2, 1)), (1, (2, 2))]
instruction = [(0, 0, 0, "uvu", True), (1, 1, 1, "uvu", True)]
# Create TP problem
tp = TPProblem(irreps_in1, irreps_in2, irreps_out, instruction, dtype=torch.float32, device='cpu')
# Input tensors
N = 8
x1 = torch.randn(N, 16)
x2 = torch.randn(N, 16)
# Forward pass
result = tp(x1, x2)
print(result.shape)
Upgrade
Version history
0.6.6latest on PyPI · released Apr 15, 2026
Audit
Dependencies
torchrequiredRequired for torch backend
jaxoptionalRequired for JAX backend