absl-py
dm-tree>=0.1.5
jax>=0.7.1
jaxlib>=0.7.1
numpy>=1.23.0

[:sys_platform != "darwin" or platform_machine != "arm64"]
tensorflow>=2.20.0

[:sys_platform == "darwin" and platform_machine == "arm64"]
tensorflow-macos>=2.20.0
