torch
torchtune
rich
lightning
torchmetrics
icecream
safetensors
wandb
torchao
transformers
deepspeed
