torch==2.9.0
torch_npu==2.9.0
triton>=3.5.0
triton-ascend==3.2.1
einops
numpy>=1.26.4
safetensors
strenum
zmq
decorator
scipy
attrs