[Torch FX] Support for ExportedProgram from torch.export.export in Weights Compression

[anzr299]

🚀 Feature request

Currently, the Torch FX backend supports FX representation(torch.fx.GraphModule() object) extracted using capture_pre_autograd_graph() which has a slightly different representation and is a deprecated feature. This feature request is to compress weights in torch FX graphs extracted using torch.export.export() instead of capture_pre_autograd_graph()

Feature Use Case

The FX model returned from torch.export.export() can also be supported in weights compression along with capture_pre_autograd_graph() due to the latter being deprecated in the future.

Are you going to submit a PR?

• [ ] Yes I'd like to help by submitting a PR!