Update requirements to include the new stable version of Tensorflow.

PiperOrigin-RevId: 391955660
Change-Id: I0800d86b5d5c9cc0612250f98f7fb7029c7216c2

requirements.txt

@@ -1,24 +1,26 @@
 absl-py==0.12.0
 atari-py==0.2.9
 bsuite==0.3.5
-dataclasses
+dataclasses==0.8;python_version=="3.6"
 dm-control==0.0.364896371
-dm-env==1.4
+dm-env==1.5
 dm-haiku==0.0.4
-dm-launchpad-nightly==0.3.0.dev20210621
-dm-reverb-nightly==0.3.0.dev20210621
+dm-launchpad-nightly==0.3.0.dev20210818
+dm-reverb==0.4.0
 dm-sonnet==2.0.0
 dm-tree==0.1.6
-jax==0.2.14
-jaxlib==0.1.67
-keras-nightly==2.6.0.dev2021062100
-optax==0.0.8
-pillow==8.2.0
+jax==0.2.17;python_version=="3.6"
+jaxlib==0.1.68;python_version=="3.6"
+jax==0.2.19;python_version>"3.6"
+jaxlib==0.1.70;python_version>"3.6"
+keras==2.6.0
+optax==0.0.9
+Pillow==8.3.1
 pytype==2021.8.11
 pytest-xdist==2.3.0
 rlax==0.0.4
-tensorflow-datasets==4.3.0
-tf-estimator-nightly==2.6.0.dev2021061601
-tf-nightly==2.6.0.dev20210621
-tfp-nightly==0.14.0.dev20210621
-trfl==1.1.0
+tensorflow-datasets==4.4.0
+tensorflow-estimator==2.6.0
+tensorflow==2.6.0
+tfp-nightly==0.14.0.dev20210818
+trfl==1.2.0

setup.py

@@ -40,20 +40,20 @@
     'jax',
     'jaxlib',
     'dm-haiku',
-    'dm-reverb-nightly',
+    'dm-reverb',
     'optax',
     'rlax',
     'dataclasses',  # Back-port for Python 3.6.
-    'tf-nightly',
+    'tensorflow',
     'tfp-nightly',
     'typing-extensions',
 ]
 
 tf_requirements = [
-    'dm-reverb-nightly',
+    'dm-reverb',
     'dm-sonnet',
     'trfl',
-    'tf-nightly',
+    'tensorflow',
     'tfp-nightly',
 ]