jax
jaxlib
flax
tqdm
requests
nmslib
numpy
transformers