torch>=2.1
numpy>=1.24
pyyaml>=6.0

[all]
composelm[datasets,dev,einops]

[datasets]
datasets>=2.14

[dev]
pytest>=7.0
pytest-cov>=4.0
ruff>=0.5
mypy>=1.8
black>=24.0

[einops]
einops>=0.7
