diff --git a/setup.py b/setup.py
index ee3c96eb1c..e673c0025f 100644
--- a/setup.py
+++ b/setup.py
@@ -36,6 +36,7 @@ setup(name='tinygrad',
             "ruff",
             "types-tqdm",
         ],
+        'mlperf': ["mlperf-logging @ git+https://github.com/mlperf/logging.git@4.0.0-rc2"],
         'testing': [
             "torch",
             "pillow",