We have pushed a second RFC PR here: Add simple `torch.mlir.export` function and demo notebook. by silvasean · Pull Request #66536 · pytorch/pytorch · GitHub to integrate with PyTorch. It follows the same api format as torch.onnx.export and provides a torch.mlir.export. If there are any suggestions / feedback on this approach from the various interested parties vs the earlier integration RFC (Add torch-mlir MLIR Exporter to PyTorch build by powderluv · Pull Request #65880 · pytorch/pytorch · GitHub) at the C++ level please let us know. The Python integration could potentially work across multiple PyTorch versions.