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

2024-04-24 23:34:34 -04:00
parent 6b8a85939d
commit d31e220cbf
1 changed files with 1 additions and 0 deletions
@@ -36,6 +36,7 @@ setup(name='tinygrad',
            "ruff",
            "types-tqdm",
        ],
+        'mlperf': ["mlperf-logging @ git+https://github.com/mlperf/[email protected]"],
        'testing': [
            "torch",
            "pillow",