add mlperf-logging to setup.py mlperf (#4289)

2026-01-09 15:08:02 -05:00 · 2024-04-24 23:34:34 -04:00
parent 6b8a85939d
commit d31e220cbf
1 changed files with 1 additions and 0 deletions
--- 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",