numpy>=1.24
torch>=2.2
safetensors>=0.4
tokenizers>=0.19
huggingface_hub>=0.23
tqdm
pyyaml

[colab]
ipywidgets
matplotlib

[datasets]
datasets>=2.14

[dev]
pytest>=7
pytest-cov
datasets>=2.14

[gpu]

[logging]
tensorboard
