jax>=0.4.0
jaxlib>=0.4.0
numpy
pandas
scipy
tqdm
