accelerate
datasets
einops
huggingface-hub
natsort
safetensors
schedulefree
simple-parsing
torch
transformers

[dev]
pre-commit
