streamlit
torch
transformers
jax
jaxlib