mosaicml[gcs,libcloud,mlflow,oci,wandb]<0.33,>=0.32.1
mlflow<3.0,>=2.14.1
accelerate<1.9,>=0.25
transformers<4.52,>=v4.51.0
mosaicml-streaming<0.13,>=0.12.0
torch<2.7.1,>=2.7.0
datasets<3.7,>=3.3.2
fsspec==2023.6.0
sentencepiece==0.2.0
einops==0.8.1
omegaconf<3,>=2.2.3
slack-sdk<4
mosaicml-cli<1,>=0.6.10
onnx==1.18.0
onnxruntime==1.22.0
boto3<2,>=1.21.45
huggingface-hub[hf_xet]<0.34,>=0.30.0
beautifulsoup4<5,>=4.12.2
tenacity<10,>=8.2.3
catalogue<3,>=2
typer<1
GitPython==3.1.44

[all]
databricks-sql-connector<4,>=3
coverage[toml]==7.9.2
pytest-cov<7,>=4
openai<2.0,>=1.56.0
pyright==1.1.256
pytest<9,>=7.2.1
pytest_codeblocks<0.18,>=0.16.1
tiktoken<0.9.1,>=0.4
mosaicml[peft]<0.33,>=0.32.1
packaging<26,>=21
flash-attn==2.7.4.post1
megablocks<1.0
pre-commit<4,>=3.4.0
hf_transfer==0.1.9
mosaicml[databricks]<0.33,>=0.32.1
mosaicml[tensorboard]<0.33,>=0.32.1
databricks-connect==14.1.0
tenacity<10,>=9
numpy<2
toml<0.11,>=0.10.2
lz4<5,>=4
grouped-gemm==0.3.0

[all-cpu]
mosaicml[peft]<0.33,>=0.32.1
coverage[toml]==7.9.2
packaging<26,>=21
databricks-connect==14.1.0
openai<2.0,>=1.56.0
tenacity<10,>=9
databricks-sql-connector<4,>=3
numpy<2
pre-commit<4,>=3.4.0
hf_transfer==0.1.9
pyright==1.1.256
toml<0.11,>=0.10.2
pytest<9,>=7.2.1
pytest_codeblocks<0.18,>=0.16.1
mosaicml[databricks]<0.33,>=0.32.1
pytest-cov<7,>=4
lz4<5,>=4
tiktoken<0.9.1,>=0.4

[all-flash2]
databricks-sql-connector<4,>=3
coverage[toml]==7.9.2
pytest-cov<7,>=4
openai<2.0,>=1.56.0
pyright==1.1.256
pytest<9,>=7.2.1
pytest_codeblocks<0.18,>=0.16.1
tiktoken<0.9.1,>=0.4
mosaicml[peft]<0.33,>=0.32.1
packaging<26,>=21
flash-attn==2.7.4.post1
megablocks<1.0
pre-commit<4,>=3.4.0
hf_transfer==0.1.9
mosaicml[databricks]<0.33,>=0.32.1
mosaicml[tensorboard]<0.33,>=0.32.1
databricks-connect==14.1.0
tenacity<10,>=9
numpy<2
toml<0.11,>=0.10.2
lz4<5,>=4
grouped-gemm==0.3.0

[databricks]
mosaicml[databricks]<0.33,>=0.32.1
numpy<2
databricks-sql-connector<4,>=3
databricks-connect==14.1.0
lz4<5,>=4

[databricks-serverless]
mosaicml[peft]<0.33,>=0.32.1
coverage[toml]==7.9.2
packaging<26,>=21
openai<2.0,>=1.56.0
tenacity<10,>=9
databricks-sql-connector<4,>=3
numpy<2
pre-commit<4,>=3.4.0
hf_transfer==0.1.9
pyright==1.1.256
toml<0.11,>=0.10.2
pytest<9,>=7.2.1
pytest_codeblocks<0.18,>=0.16.1
mosaicml[databricks]<0.33,>=0.32.1
pytest-cov<7,>=4
lz4<5,>=4
tiktoken<0.9.1,>=0.4

[dev]
coverage[toml]==7.9.2
pre-commit<4,>=3.4.0
pytest<9,>=7.2.1
pytest_codeblocks<0.18,>=0.16.1
pytest-cov<7,>=4
pyright==1.1.256
toml<0.11,>=0.10.2
packaging<26,>=21
hf_transfer==0.1.9
tenacity<10,>=9

[gpu]
flash-attn==2.7.4.post1

[gpu-flash2]
flash-attn==2.7.4.post1

[megablocks]
megablocks<1.0
grouped-gemm==0.3.0

[openai]
openai<2.0,>=1.56.0
tiktoken<0.9.1,>=0.4

[peft]
mosaicml[peft]<0.33,>=0.32.1

[te]
transformer-engine[pytorch]<2.0,>=1.13.0

[tensorboard]
mosaicml[tensorboard]<0.33,>=0.32.1
