transformers
accelerate
datasets
torchmetrics
torchmultimodal-nightly
einops
wandb
yacs
