torch>=2.1.0

[all]
datasets>=2.16
transformers>=4.38
tqdm
tensorboard
pytest>=7.0
pytest-cov

[dev]
pytest>=7.0
pytest-cov

[train]
datasets>=2.16
transformers>=4.38
tqdm
tensorboard
