torch>=2.1
einops
onnxscript
onnx
packaging
pydantic
nvdlfw-inspect
nvidia-cudnn-frontend>=1.29.0
transformer_engine_cu13==2.20.2
