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

This commit is contained in:
chenyu 2024-04-24 23:34:34 -04:00 committed by GitHub
commit d31e220cbf
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -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",