torch
torchvision
peft
wandb
datasets
transformers
scipy
scikit-learn