torch>=2.8
numpy>=1.24
pyyaml>=6.0
safetensors>=0.4

[all]
composelm[datasets,dev,einops,stateful-data,tracking]

[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

[stateful-data]
torchdata>=0.8

[tracking]
tensorboard>=2.12
wandb>=0.16
