numpy
sentencepiece
tfds-nightly
tensorflow-gpu==2.2.0
tensorflow==2.2.0