It would be great if the PyTorch version that was used was mentioned here as well. I was stuck because my trace was not displaying CompiledFxGraph. It worked on Colab but not on my local machine. I was using PyTorch 2.8.0, which does not show CompiledFxGraph in the trace.
Thanks for the awesome blog!