streamlit
torch
transformers
jax