streamlit
torch
transformers[sentencepiece]
jax
jaxlib
flax