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

[all]
transformers<5.0.0,>=4.35.0
datasets<3.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<1.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<6.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<3.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<6.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
pre-commit<4.0.0,>=3.5.0
bandit[toml]<2.0.0,>=1.7.0

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

[monitoring]
prometheus-client<1.0.0,>=0.17.0
wandb<1.0.0,>=0.16.0
tensorboard<3.0.0,>=2.14.0
mlflow<3.0.0,>=2.7.0
optuna<4.0.0,>=3.4.0

[quantization]
torchao<1.0.0,>=0.4.0

[serving]
fastapi<1.0.0,>=0.103.0
uvicorn[standard]<1.0.0,>=0.23.0
torchserve<1.0.0,>=0.8.0
gradio<6.0.0,>=3.45.0
streamlit<2.0.0,>=1.27.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
