torch>=2.1
einops
onnxscript
onnx
packaging
pydantic
nvdlfw-inspect
transformer_engine_rocm7==2.14.0.dev0+rocm7.14.0a20260620
