torch
numpy
pandas
lightning
sentencepiece