ONNX Export for multiple input / output types
The ONNX model should eventually be able to support multiple input types. We would also need to make sure that the auxiliary tasks know which input type to decorate their outputs to.
We might need to add this to our custom json configuration, I'm not sure we could put it directly on the ONNX model itself (though we should certainly try).
This was discussed a bit here atlas/athena!66718 (comment 7242673)