einops

[cpu]
torch>=2.7.0
triton>=3.3

[cuda]
torch>=2.7.0
triton>=3.3

[npu]
torch==2.10.0
torch_npu==2.10.0.post7
torchvision==0.25.0
triton-ascend==3.6.0

[rocm]
torch>=2.7.0

[xpu]
torch>=2.7.0
