How to replace modules like TorchFX for compiled model

[cokespace2 (Vince Mo)]

In TorchFX we can replace modules after symbolic trace returning a fx.GraphModule, how can we do that for a compiled module from torch.compile?

[marksaroufim]

It’s quite similar this notebook should give you more ideas as how this would work Google Colab