numpy
torch
transformers
ujson
accelerate