mosaicml[nlp,streaming,wandb]<0.15,>=0.14.1
torch==1.13.1
datasets==2.10.1
sentencepiece==0.1.97
einops==0.5.0
omegaconf<3,>=2.2.3
pynvml<12
slack-sdk<4
mosaicml-cli<1,>=0.3
onnx==1.13.1
onnxruntime==1.14.1

[all]
pytest-cov<5,>=4
pre-commit<3,>=2.18.1
pyright==1.1.296
packaging<23,>=21
pytest<8,>=7.2.1
toml<0.11,>=0.10.2
triton==2.0.0.dev20221202
flash-attn==v1.0.3.post0
pytest_codeblocks<0.17,>=0.16.1

[dev]
pre-commit<3,>=2.18.1
pytest<8,>=7.2.1
pytest_codeblocks<0.17,>=0.16.1
pytest-cov<5,>=4
pyright==1.1.296
toml<0.11,>=0.10.2
packaging<23,>=21

[gpu]
flash-attn==v1.0.3.post0
triton==2.0.0.dev20221202
