dm-haiku==0.0.13
ipykernel==6.29.5
ipynbname==2024.1.0.0
ipython==8.30.0
jax[cuda12]==0.4.35
jaxlib==0.4.34
keras==3.7.0
mctx==0.0.5
numpy==2.0.2
omegaconf==2.3.0
optax==0.2.4
pgx==2.5.0
pydantic==2.10.3
pytz==2024.2
scipy==1.14.1
wandb==0.19.0
