numpy
pandas
scipy
torch
scikit-learn
tqdm
datasets
gymnasium
transformers[torch]
datasets
huggingface-hub
