chex>=0.0.4
cloudpickle>=1.2.2
dill
mock>=3.0.5
pytest-xdist
virtualenv
dm-tree>=0.1.1
dataclasses==0.7 ; python_version<'3.7'
tf-nightly==2.7.0.dev20210906
optax==0.0.1

chex>=0.0.4
cloudpickle>=1.2.2
dill
mock>=3.0.5
pytest-xdist
virtualenv
dm-tree>=0.1.1
dataclasses==0.7 ; python_version<'3.7'
tensorflow==2.7.0
optax==0.0.1

Replace tf-nightly with tensorflow==2.7.0 in tests.
Replace tf-nightly with tensorflow==2.7.0 in tests.

PiperOrigin-RevId: 407787866
