numpy
jax
tqdm
