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 39 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 19) at the C++ level please let us know.  The Python integration could potentially work across multiple PyTorch versions.