torch
transformers
tokenizers
datasets
peft
wandb
loguru
nvitop
lion-pytorch
