torch>=2.6

[dev]
pytest
ruff

[train]
transformers
datasets
