streamlit
transformers
jax
flax
torch