Assuming you're talking about the one here: https://pytorch.org/blog/accelerating-generative-ai-2/#starting-point-255-toks https://pytorch.org/blog/accelerating

Assuming you're talking about the one here: https://pytorch.org/blog/accelerating-generative-ai-2/#starting-point-255-toks https://pytorch.org/blog/accelerating-generative-ai-2/#start... it's just the pytorch profiler + chrome profiler (chrome://tracing)