Посмотрите на профайлер tinygrad с VIZ=1 (это профилирование стабильной диффузии). Просто `VIZ=1`, и он загружается в браузере, никаких настроек или импортов не требуется.