torch
transformers
datasets
peft
trl