setuptools
tqdm
torch
einops
flash_attn>2.6.3
accelerate
pandas
numpy
polars
torchmetrics
lightning
scikit-learn
huggingface_hub
