aqtp  # for flaxformer
einops
flax
gradio
jax
jaxlib
ml_collections
numpy
Pillow
sentencepiece
transformers  # for transformers.BertTokenizer
tensorflow  # for tf.io.gfile