loguru
trl
transformers
datasets
tqdm

[dev]
torch

[test]
torch
pytest
pytest-xdist
