einops

[cpu]
torch>=2.7.0
triton>=3.3

[cuda]
torch>=2.7.0
triton>=3.3

[npu]
torch==2.7.1
torch_npu==2.7.1.post4
torchvision==0.22.1
triton-ascend==3.2.1

[rocm]
torch>=2.7.0

[xpu]
torch>=2.7.0
