git+https://github.com/deepmind/dm-haiku
jax~=0.2.11
jaxlib~=0.1.64
matplotlib
numpy
optax
scipy
simple-parsing
tqdm