daquexian / onnx-simplifier

Simplify your onnx model
Apache License 2.0
3.79k stars 380 forks source link

Feature Request: Rust Wrapper for onnx-simplifier for Streamlined ONNX Parsing #292

Open antimora opened 1 year ago

antimora commented 1 year ago

I am a part of the Burn team, and we are currently developing a Deep Learning Framework in Rust.

We've been investing significant efforts in our Burn Import project, which generates Rust code for Burn from ONNX files. Dealing with the complexity of ONNX, as you may well understand, is quite a challenge. Thus, we would like to integrate onnx-simplifier into our pipeline to simplify the ONNX files before we parse them.

Rather than asking users to manually run onnx-simplifier on their ONNX files, we believe it would be more efficient to automate this process. Given that you've successfully wrapped your C++ code with Python, we are optimistic that a similar accomplishment could be possible with Rust.

We are not alone in this: other Rust teams, such as wonnx and tract, would also likely benefit from this functionality.

daquexian commented 11 months ago

Sorry for the late response. It will be definitely amazing if we can have a Rust binding, but I'm not familiar with Rust. Would you like to propose a pull request about it?