The nuances of PyTorch Graph Capture

No, torch.jit.script() and torch.jit.trace() are not yet deprecated because of two reasons:

  • Many vendors are still integrating their backends via TorchScript IR
  • torch.export() is not yet ready

We do want to consolidate front-end solutions into TorchDynamo organically. The speed of this consolidation depends on how fast vendors integrate to PT2 stack and how soon we have PT2 export ready.

1 Like