torch>=2.7.0
torchvision
triton>=3.3.0
flash-attn
einops
numpy
pillow
loguru
imageio[ffmpeg]
pandas
PyYAML
omegaconf
attrs
fvcore
ftfy
regex
transformers
nvidia-ml-py
