torch>=2.6.0
transformer_engine[pytorch]==1.13.0
numpy
einops==0.8.0
packaging
rich
tqdm
PyYAML
