Ознайомтеся з профайлером tinygrad з VIZ=1 (це профілювання стабільної дифузії). Просто 'VIZ=1' і він завантажується в браузері, потрібно налаштувати 0 або імпортувати.