torch>=2.0
nvidia-ml-py>=12.0
numpy>=1.24
scipy>=1.10
psutil>=5.9
dataclasses-json>=0.6
rich>=13.0

[deepspeed]
deepspeed>=0.12

[dev]
pytest>=7.0
pytest-cov>=4.0
pytest-mock>=3.10
hypothesis>=6.80
black>=23.0
ruff==0.15.8
mypy>=1.5
types-psutil

[hf]
transformers>=4.35
accelerate>=0.24
