torch
onnxruntime-gpu
numpy
sentencepiece