diff --git a/setup.py b/setup.py
index 199ef5ff..160e7ddb 100644
--- a/setup.py
+++ b/setup.py
@@ -103,7 +103,7 @@ def build_assets(self):
     python_requires='>=3.11',
     install_requires=[
         'absl-py',
-        'chex<0.1.81',  # Incompatible with tensorflow 2.13 (due to numpy req).
+        'chex',
         'dm-env',
         'dmlab2d',
         'dm-tree',