trl
transformers
sentencepiece
tyro
jsonargparse
deepspeed
tqdm
rich
wandb
scikit-learn  # eval
pandas  # stats
matplotlib  # stats - plots
ipdb
flash-attn  # pay attention to the CUDA and PyTorch version for pre-compiled wheel!
