transformers
datasets
peft
scipy
scikit-learn
trl
tensorboard
array_api_compat