torch
torchvision
trl
transformers
accelerate
peft
einops
datasets
bitsandbytes
scipy