einops==0.8.0
torch==2.3.0
mosaicml[nlp,wandb]>=0.22.0,<0.23
mosaicml-streaming==0.7.6
omegaconf==2.3.0
transformers==4.40.2
triton==2.3.0
