Echa un vistazo al perfilador de tinygrad con VIZ=1 (esto es perfilar la difusión estable). Simplemente 'VIZ=1' y se carga en un navegador, se requiere 0 configuración o importaciones.