Skip to content

Commit

Permalink
Update note about torchxla2
Browse files Browse the repository at this point in the history
  • Loading branch information
pavithraes committed Nov 11, 2024
1 parent 502a0aa commit 2e42f09
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/tutorials/pytorch-export.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -139,7 +139,7 @@
"\n",
"_Dynamic batch dimensions can be specified as a part of the initial `torch.export` step._\n",
"\n",
"_`torch_xla` doesn't yet support exporting models with dynamic shapes to StableHLO, so we recommend using the experimental library [`torch_xla2`](https://github.com/pytorch/xla/tree/master/experimental/torch_xla2) for this. It has a similar API for exporting, converts PyTorch models to JAX prior to exporting to StableHLO, and has high opset coverage along with great dynamic shape support._"
"_`torch_xla`'s support for exporting dynamic models is limited, for these cases we recommend using [`torch_xla2`](https://github.com/pytorch/xla/tree/master/experimental/torch_xla2) for this. This lowering path leverages JAX for lowering to StableHLO, and has high opset coverage with much broader support for exported programs with dynamic shapes._"
]
},
{
Expand Down

0 comments on commit 2e42f09

Please sign in to comment.