jax
jaxlib
flax
transformers
streamlit
tqdm
nmslib
matplotlib