transformers
torch
torch-scatter -f https://data.pyg.org/whl/torch-1.10.0+cu102.html
pandas
numpy
tensorflow