transformers[torch]
torch

[dev]
ruff
pytest
pytest-xdist

[fast]
numba

[train]
datasets
evaluate
scikit-learn
wandb
