einops
numpy
seaborn
torch
transformers
