transformers 
datasets 
accelerate 
nvidia-ml-py3
sentencepiece