flash-attn
sentencepiece>=0.2.0
tensorboardX
