torch
transformers
tokenizers
numpy