diff --git a/setup.py b/setup.py
index 803f193044..af12147276 100644
--- a/setup.py
+++ b/setup.py
@@ -21,4 +21,12 @@ setup(name='tinygrad',
       ],
       install_requires=['numpy', 'requests'],
       python_requires='>=3.8',
+      extras_require={
+        'gpu': ["pyopencl"],
+        'testing': [
+            "pytest",
+            "torch",
+            "tqdm",
+        ],
+      },
       include_package_data=True)