optimizer
optimizer copied to clipboard
[Feature request] Integrate Python-only API to ONNX optimizer proposal
Hello @daquexian, I am with Microsoft developing the PyTorch ONNX Exporter project.
Currently we are working on a PyTorch Dynamo-based ONNX exporter (torch.onnx.dynamo_export) which will replace the existing Torchscript-based ONNX exporter (torch.onnx.export) once it is production-ready.
Due to how Dynamo works, the ONNX graphs produced doesn't really perform well as the ones acquired through the previous exporter. As a response, we are internally working on an ONNX optimizer API which is Python only. As the workstream approaches stability, we are looking into how to open source the library.
One of the possibilities would be to create a new library, but we are also interested in maybe merging the new APIs in this repository. The goals of our library and ONNX optimizer's align well and the new implementation might help growing the project.
Is this something ONNX org would be interested at? We can discuss more about what it would encompasses the new API, etc
Thanks