chex>=0.0.8
dm-haiku>=0.0.4
dm-sonnet>=2.0.0
tensorflow==2.9.1; sys_platform != 'darwin' or platform_machine != 'arm64'
tensorflow-macos==2.9.1; sys_platform == 'darwin' and platform_machine == 'arm64'
