torch>=2.1
einops
onnxscript
onnx
transformer_engine_rocm7==2.10.0+rocm7.15.0a20260628.b2a0450
