Someone please port this to ONNX so we don't need to do all this ass tooling