torch
transformers
tokenizers
datasets
peft
wandb
loguru
nvitop
lion-pytorch
matplotlib
scipy
scikit-learn
evaluate
tensorly
