torch>=2.3.0
torchvision
torchaudio
deepspeed
hydra-core==1.3.2
omegaconf==2.3.0
einops==0.8.0
jaxtyping==0.2.28
transformers==4.41.2
accelerate==0.30.1
av==10.0.0
ninja==1.11.1.1
wandb
webdataset
matplotlib
black
pre-commit
torchcodec
