dm-haiku>=0.0.6
jax>=0.3.13
optax>=0.1.2
jmp>=0.0.2
chex>=0.1.3
keras>=2.9.0
tensorflow>=2.9.0
rich>=12.4.1
