numpy<1.26,>=1.25
pandas==1.5.3
jax>=0.4.31
jaxlib>=0.4.31
flax>=0.7.4
optax>=0.1.7
tensorflow_probability==0.23.0
chex>=0.1.83
wandb>=0.13
ipython>=7.34.0
GitPython>=3.1.29
tqdm>=4.66.1
