Context
README says: “This currently works on PyTorch 2.8.0.dev20250506.”
I tried to reproduce the AutoParallel workflow (capture joint fwd+bwd → feed solver) following this guidance.
What I tried (in order)
Recommended version
torch==2.8.0.dev20250506 is no longer obtainable from the nightly index (pip can’t find it; direct wheel URL returns 403).
Nearby versions (built from source)
I built nightlies close to 2.8.0.dev20250506 successfully.
When running AutoParallel:
I first hit the same error:
RuntimeError: Joint export did not yield a torch.fx.GraphModule（same as issue #241 but the answer there can't help）.
After patching/workarounds, I then kept hitting new ModuleNotFoundError at runtime (cascading missing modules).
I fixed 3–4 of these, but each fix exposed another missing module, so this path did not converge.
Latest nightly
On PyTorch 2.11.0.dev20260121, joint export still consistently fails with:
RuntimeError: Joint export did not yield a torch.fx.GraphModule.
This happens for both torch.export.aot_export_joint_with_descriptors and the functorch fallback. A best-effort recursive extraction over the returned object still finds no torch.fx.GraphModule.
Conclusion / Question
At the moment, I can’t find a PyTorch version that is both:
obtainable/reproducible, and
compatible with AutoParallel’s current FX GraphModule requirement.
Could you:
Point to a currently obtainable PyTorch version known to work?
Or clarify whether AutoParallel is expected to support the new torch.export / ExportedProgram IR instead of FX?
Context
README says: “This currently works on PyTorch 2.8.0.dev20250506.”
I tried to reproduce the AutoParallel workflow (capture joint fwd+bwd → feed solver) following this guidance.
What I tried (in order)
Recommended version
torch==2.8.0.dev20250506 is no longer obtainable from the nightly index (pip can’t find it; direct wheel URL returns 403).
Nearby versions (built from source)
I built nightlies close to 2.8.0.dev20250506 successfully.
When running AutoParallel:
I first hit the same error:
RuntimeError: Joint export did not yield a torch.fx.GraphModule（same as issue #241 but the answer there can't help）.
After patching/workarounds, I then kept hitting new ModuleNotFoundError at runtime (cascading missing modules).
I fixed 3–4 of these, but each fix exposed another missing module, so this path did not converge.
Latest nightly
On PyTorch 2.11.0.dev20260121, joint export still consistently fails with:
RuntimeError: Joint export did not yield a torch.fx.GraphModule.
This happens for both torch.export.aot_export_joint_with_descriptors and the functorch fallback. A best-effort recursive extraction over the returned object still finds no torch.fx.GraphModule.
Conclusion / Question
At the moment, I can’t find a PyTorch version that is both:
obtainable/reproducible, and
compatible with AutoParallel’s current FX GraphModule requirement.
Could you:
Point to a currently obtainable PyTorch version known to work?
Or clarify whether AutoParallel is expected to support the new torch.export / ExportedProgram IR instead of FX?