torch<2.11.0,>=2.10.0
torchvision<0.26.0,>=0.25.0
einops>=0.6.0
numpy>=1.21.0
pyarrow<20.0.0,>=14.0.0
datasets>=2.14.0
megatron-core
pytorch-lightning>=2.0.0
omegaconf>=2.3.0
wandb>=0.15.0
protobuf>=4.23.0
rich>=13.0.0
Pillow
huggingface_hub
subquadratic-ops-torch-cu12>=0.2.0
timm==1.0.24
nvidia-dali-cuda120==1.53.0

[quack]
quack-kernels
