torch>=2.1
einops
onnxscript
onnx
packaging
pydantic
transformer_engine_cu12==2.11.0
