chex>=0.0.8
dm-haiku>=0.0.4
dm-sonnet>=2.0.0
tensorflow==v2.9.1
