mosaicml[gcs,libcloud,mlflow,oci,wandb]<0.24,>=0.23.4
mlflow<2.16,>=2.14.1
accelerate<0.34,>=0.25
transformers<4.44,>=4.43.2
mosaicml-streaming<0.9,>=0.8.0
torch<2.4,>=2.3.0
datasets<2.20,>=2.19
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.16.2
onnxruntime==1.18.1
boto3<2,>=1.21.45
huggingface-hub<0.25,>=0.19.0
beautifulsoup4<5,>=4.12.2
tenacity<9,>=8.2.3
catalogue<3,>=2
typer<1

[all]
mosaicml[tensorboard]<0.24,>=0.23.4
pytest-cov<6,>=4
tiktoken==0.4.0
hf_transfer==0.1.3
packaging<23,>=21
openai==1.3.8
grouped-gemm==0.1.4
pyright==1.1.256
databricks-sql-connector<4,>=3
coverage[toml]==7.4.4
toml<0.11,>=0.10.2
pre-commit<4,>=3.4.0
flash-attn<3,>=2.5.8
databricks-connect==14.1.0
mosaicml[peft]<0.24,>=0.23.4
pytest_codeblocks<0.17,>=0.16.1
mosaicml[databricks]<0.24,>=0.23.4
megablocks==0.5.1
lz4<5,>=4
pytest<8,>=7.2.1

[all-cpu]
pytest_codeblocks<0.17,>=0.16.1
databricks-sql-connector<4,>=3
coverage[toml]==7.4.4
lz4<5,>=4
hf_transfer==0.1.3
mosaicml[tensorboard]<0.24,>=0.23.4
toml<0.11,>=0.10.2
packaging<23,>=21
mosaicml[databricks]<0.24,>=0.23.4
openai==1.3.8
pre-commit<4,>=3.4.0
pyright==1.1.256
tiktoken==0.4.0
pytest-cov<6,>=4
databricks-connect==14.1.0
mosaicml[peft]<0.24,>=0.23.4
pytest<8,>=7.2.1

[all-flash2]
mosaicml[tensorboard]<0.24,>=0.23.4
pytest-cov<6,>=4
tiktoken==0.4.0
hf_transfer==0.1.3
packaging<23,>=21
openai==1.3.8
grouped-gemm==0.1.4
pyright==1.1.256
databricks-sql-connector<4,>=3
coverage[toml]==7.4.4
toml<0.11,>=0.10.2
pre-commit<4,>=3.4.0
flash-attn<3,>=2.5.8
databricks-connect==14.1.0
mosaicml[peft]<0.24,>=0.23.4
pytest_codeblocks<0.17,>=0.16.1
mosaicml[databricks]<0.24,>=0.23.4
megablocks==0.5.1
lz4<5,>=4
pytest<8,>=7.2.1

[databricks]
mosaicml[databricks]<0.24,>=0.23.4
databricks-sql-connector<4,>=3
databricks-connect==14.1.0
lz4<5,>=4

[databricks-serverless]
pytest_codeblocks<0.17,>=0.16.1
databricks-sql-connector<4,>=3
coverage[toml]==7.4.4
lz4<5,>=4
hf_transfer==0.1.3
mosaicml[tensorboard]<0.24,>=0.23.4
toml<0.11,>=0.10.2
packaging<23,>=21
mosaicml[databricks]<0.24,>=0.23.4
openai==1.3.8
pre-commit<4,>=3.4.0
pyright==1.1.256
tiktoken==0.4.0
pytest-cov<6,>=4
mosaicml[peft]<0.24,>=0.23.4
pytest<8,>=7.2.1

[dev]
coverage[toml]==7.4.4
pre-commit<4,>=3.4.0
pytest<8,>=7.2.1
pytest_codeblocks<0.17,>=0.16.1
pytest-cov<6,>=4
pyright==1.1.256
toml<0.11,>=0.10.2
packaging<23,>=21
hf_transfer==0.1.3

[gpu]
flash-attn<3,>=2.5.8

[gpu-flash2]
flash-attn<3,>=2.5.8

[megablocks]
megablocks==0.5.1
grouped-gemm==0.1.4

[openai]
openai==1.3.8
tiktoken==0.4.0

[peft]
mosaicml[peft]<0.24,>=0.23.4

[tensorboard]
mosaicml[tensorboard]<0.24,>=0.23.4
