numpy
datasets
lightning
matplotlib
scikit-learn
scipy
torch
torchmetrics
transformers
torchvision
loralib
wandb
tensorboard
tensorboardX
peft
termcolor
bitsandbytes