Once https://github.com/huggingface/optimum/pull/226 is merged it would be good to explore using optimum/onnx again