apache-beam
google-vizier>=0.0.13
ml-collections
networkx
pyarrow

[:platform_machine != "arm64" or platform_system != "Darwin"]
tensorflow>=2.8.0

[:platform_machine == "arm64" and platform_system == "Darwin"]
tensorflow-macos>=2.8.0
