torch>=2.1
einops
onnxscript
onnx
transformer_engine_rocm7==2.10.0+devrocm7.14.0.dev0.22b25e93720109d1f2e3b475df7c89cfd29aad98.b2a0450
