torch>=2.0.0
numpy>=1.21.0
pybind11>=2.10.0

[all]
transformers>=4.35.0
datasets>=2.14.0
tokenizers>=0.14.0
triton>=2.1.0
flash-attn>=2.3.0
accelerate>=0.24.0

[benchmark]
memory-profiler>=0.61.0
py-spy>=0.3.14
torch-tb-profiler>=0.4.0
psutil>=5.9.0
gpustat>=1.1.0

[cloud]
boto3>=1.28.0
google-cloud-storage>=2.10.0
azure-storage-blob>=12.17.0
kubernetes>=27.2.0

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0
pytest-xdist>=3.0.0
pytest-benchmark>=4.0.0
matplotlib>=3.5.0
seaborn>=0.11.0
jupyter>=1.0.0
tensorboard>=2.9.0
ruff>=0.4.0
mypy>=1.0.0
pre-commit>=3.5.0
bandit[toml]>=1.7.0

[monitoring]
prometheus-client>=0.17.0
wandb>=0.16.0
tensorboard>=2.14.0
mlflow>=2.7.0
optuna>=3.4.0

[serving]
fastapi>=0.103.0
uvicorn[standard]>=0.23.0
torchserve>=0.8.0
gradio>=3.45.0
streamlit>=1.27.0
