Skip to content

Support the new PyTorch torch.export path (since jit.trace is discontinued) #15

Description

@JulienBalianSonos
  • Starting with PyTorch 2.9, jit.trace is no longer maintained.
  • The goal of this effort is to port to the new torch.export path.

This will require a significant refactor of the torch_graph module to replace the current logic. It may enable better analysis of symbolic dimensions, allowing more fine-grained serialization (by resolving constant dimensions directly when known as invariant at export time).

torch.jit has been tested and still works, but a bug has already surfaced with the conjunction of Python 3.14 and PyTorch >=2.9, causing RAM usage to explode (details).

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Fields

    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions