Files
tinygrad/extra/optimization/extract_policynet.py