torch>=2.1.0
numpy>=1.21.0
psutil>=5.9.0

[all]
memscale[hf,lightning,observability]

[dev]
pytest>=7.0
pytest-cov>=4.0
black>=23.0
ruff>=0.1.0
mypy>=1.0

[hf]
transformers>=4.30.0

[lightning]
lightning>=2.0.0

[observability]
prometheus-client>=0.17.0
tqdm>=4.65.0
