torch
transformers
jax
flax
altair==4.0
streamlit