numpy
pandas
matplotlib
scikit-learn
jupyter
tqdm

jax==0.2.10
jaxlib

