huggingface-hub
transformers # 4.38.0 for TrainingArguments, 4.55.4 for TRL
polars>1
numpy
seaborn
torch
wandb
datasets
accelerate
evaluate
scikit-learn
pynvml
# torchtune
# torchao
peft
trl
bitsandbytes