mosaicml[gcs,libcloud,mlflow,oci,wandb]<0.29,>=0.28.0
mlflow<2.19,>=2.14.1
accelerate<1.2,>=0.25
transformers<4.47,>=4.43.2
mosaicml-streaming<0.12,>=0.11.0
torch<2.5.2,>=2.5.1
datasets<3.3,>=2.20.0
fsspec==2023.6.0
sentencepiece==0.2.0
einops==0.8.0
omegaconf<3,>=2.2.3
slack-sdk<4
mosaicml-cli<1,>=0.6.10
onnx==1.17.0
onnxruntime==1.19.2
boto3<2,>=1.21.45
huggingface-hub<0.27,>=0.19.0
beautifulsoup4<5,>=4.12.2
tenacity<10,>=8.2.3
catalogue<3,>=2
typer<1
GitPython==3.1.43

[all]
megablocks<1.0
grouped-gemm==0.1.6
pytest-cov<7,>=4
transformer-engine[pytorch]<1.12,>=1.11.0
databricks-sql-connector<4,>=3
mosaicml[peft]<0.29,>=0.28.0
openai<2.0,>=1.56.0
packaging<25,>=21
toml<0.11,>=0.10.2
lz4<5,>=4
mosaicml[tensorboard]<0.29,>=0.28.0
pre-commit<4,>=3.4.0
pytest_codeblocks<0.18,>=0.16.1
pytest<9,>=7.2.1
databricks-connect==14.1.0
flash-attn==2.6.3
numpy<2
mosaicml[databricks]<0.29,>=0.28.0
pyright==1.1.256
coverage[toml]==7.6.10
hf_transfer==0.1.8
tiktoken<0.8.1,>=0.4

[all-cpu]
packaging<25,>=21
lz4<5,>=4
numpy<2
pytest-cov<7,>=4
databricks-sql-connector<4,>=3
mosaicml[peft]<0.29,>=0.28.0
mosaicml[databricks]<0.29,>=0.28.0
pyright==1.1.256
pre-commit<4,>=3.4.0
coverage[toml]==7.6.10
openai<2.0,>=1.56.0
pytest_codeblocks<0.18,>=0.16.1
hf_transfer==0.1.8
pytest<9,>=7.2.1
toml<0.11,>=0.10.2
databricks-connect==14.1.0
tiktoken<0.8.1,>=0.4

[all-flash2]
megablocks<1.0
grouped-gemm==0.1.6
pytest-cov<7,>=4
transformer-engine[pytorch]<1.12,>=1.11.0
databricks-sql-connector<4,>=3
mosaicml[peft]<0.29,>=0.28.0
openai<2.0,>=1.56.0
packaging<25,>=21
toml<0.11,>=0.10.2
lz4<5,>=4
mosaicml[tensorboard]<0.29,>=0.28.0
pre-commit<4,>=3.4.0
pytest_codeblocks<0.18,>=0.16.1
pytest<9,>=7.2.1
databricks-connect==14.1.0
flash-attn==2.6.3
numpy<2
mosaicml[databricks]<0.29,>=0.28.0
pyright==1.1.256
coverage[toml]==7.6.10
hf_transfer==0.1.8
tiktoken<0.8.1,>=0.4

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

[databricks-serverless]
packaging<25,>=21
lz4<5,>=4
numpy<2
pytest-cov<7,>=4
databricks-sql-connector<4,>=3
mosaicml[peft]<0.29,>=0.28.0
mosaicml[databricks]<0.29,>=0.28.0
pyright==1.1.256
pre-commit<4,>=3.4.0
coverage[toml]==7.6.10
openai<2.0,>=1.56.0
pytest_codeblocks<0.18,>=0.16.1
hf_transfer==0.1.8
pytest<9,>=7.2.1
toml<0.11,>=0.10.2
tiktoken<0.8.1,>=0.4

[dev]
coverage[toml]==7.6.10
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<25,>=21
hf_transfer==0.1.8

[gpu]
flash-attn==2.6.3

[gpu-flash2]
flash-attn==2.6.3

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

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

[peft]
mosaicml[peft]<0.29,>=0.28.0

[te]
transformer-engine[pytorch]<1.12,>=1.11.0

[tensorboard]
mosaicml[tensorboard]<0.29,>=0.28.0
