torch<3.0.0,>=2.0.0
numpy<3.0.0,>=1.21.0

[all]
transformers<5.0.0,>=4.35.0
datasets<5.0.0,>=2.14.0
tokenizers<1.0.0,>=0.14.0
triton<4.0.0,>=2.1.0
flash-attn<4.0.0,>=2.3.0
accelerate<2.0.0,>=0.24.0

[benchmark]
memory-profiler<1.0.0,>=0.61.0
py-spy<1.0.0,>=0.3.14
torch-tb-profiler<1.0.0,>=0.4.0
psutil<8.0.0,>=5.9.0
gpustat<2.0.0,>=1.1.0

[checkpoint]
s3fs<2026.0.0,>=2024.1.0
gcsfs<2026.0.0,>=2024.1.0
adlfs<2026.0.0,>=2024.1.0

[cloud]
boto3<2.0.0,>=1.28.0
google-cloud-storage<4.0.0,>=2.10.0
azure-storage-blob<13.0.0,>=12.17.0
kubernetes<32.0.0,>=27.2.0

[dev]
pytest<9.0.0,>=7.0.0
pytest-cov<8.0.0,>=4.0.0
pytest-xdist<4.0.0,>=3.0.0
pytest-benchmark<5.0.0,>=4.0.0
hypothesis<7.0.0,>=6.0.0
matplotlib<4.0.0,>=3.5.0
seaborn<1.0.0,>=0.11.0
jupyter<2.0.0,>=1.0.0
tensorboard<3.0.0,>=2.9.0
ruff<1.0.0,>=0.4.0
mypy<2.0.0,>=1.0.0
types-protobuf<8.0.0,>=4.24.0
pre-commit<5.0.0,>=3.5.0
bandit[toml]<2.0.0,>=1.7.0

[docs]
sphinx<9.0.0,>=7.0.0
furo<2026.0.0,>=2024.1.0
myst-parser<5.0.0,>=2.0.0
sphinx-autobuild<2026.0.0,>=2024.1.0

[monitoring]
wandb<1.0.0,>=0.16.0
tensorboard<3.0.0,>=2.14.0
mlflow<3.0.0,>=2.7.0
optuna<5.0.0,>=3.4.0

[quantization]
torchao<1.0.0,>=0.4.0

[tracing]
opentelemetry-api<2.0.0,>=1.20.0
opentelemetry-sdk<2.0.0,>=1.20.0
opentelemetry-exporter-otlp-proto-http<2.0.0,>=1.20.0
