Files
tinygrad/extra/optimization/extract_dataset.py