setuptools
tqdm
torch
einops
flash_attn
accelerate
pandas
numpy
polars
torchmetrics
lightning
