chex>=0.0.8
dm-haiku>=0.0.4
dm-sonnet>=2.0.0
tensorflow==v2.9.0-rc1 # jax2tf at jax>=0.3.8 requires tf-nightly or tensorflow>=2.9.0
