transformers
torch
datasets
Cython==0.29.21
IPython
wandb
accelerate 
onnx