absl-py
chex
distrax>=0.1.2
dm-haiku
flax
jax
matplotlib
ml-collections
numpy
optax
pandas

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

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