nvidia-ml-py
psutil
textual
rich
pyyaml
numpy
scikit-learn
prometheus-client
cryptography

[accelerate]
accelerate>=0.20.0
torch>=2.0.0

[all]
torch>=2.0.0
pytorch-lightning>=2.0.0
deepspeed>=0.10.0
accelerate>=0.20.0
ray[train]>=2.0.0
mlflow>=2.0.0
wandb>=0.15.0
kubernetes>=27.0.0
pydantic>=2.0.0

[deepspeed]
deepspeed>=0.10.0
torch>=2.0.0

[dev]
pytest
pytest-asyncio
pytest-cov

[kubernetes]
kubernetes>=27.0.0

[lightning]
pytorch-lightning>=2.0.0
torch>=2.0.0

[mlflow]
mlflow>=2.0.0

[pytorch]
torch>=2.0.0

[ray]
ray[train]>=2.0.0

[validation]
pydantic>=2.0.0

[wandb]
wandb>=0.15.0
