pytorch - 💡(How to fix) Fix [Question] JIT Path Deprecation — Maintenance Timeline and Removal Plans [1 participants]

Fix / Guidance: Keep using TorchScript (JIT) while migrating to torch.compile

The JIT (TorchScript) API is still fully supported in the current stable release. The “deprecated” notice you saw refers only to the documentation page being moved, not to the runtime itself. No removal is planned in the near‑term, but the community is encouraging new projects to adopt the newer torch.compile workflow for better performance and easier maintenance.

Below is a quick migration path you can follow today, with concrete code snippets.

1. Verify your current JIT usage

import torch

class MyModel(torch.nn.Module):
    def __init__(self):
        super().__init__()
        self.linear = torch.nn.Linear(10, 1)

    def forward(self, x):
        return self.linear(x)

model = MyModel()
example = torch.randn(1, 10)

# Existing TorchScript (JIT) workflow
scripted = torch.jit.script(model)          # or torch.jit.trace(model, example)
torch.save(scripted.state_dict(), "model_jit.pt")

Run the script and confirm it works on your target platform.

2. Add a torch.compile fallback (no code change needed)

# New recommended path