ONNX IR supports all ONNX dtypes and provides a easier and more robust interfaces for model transformation. I suggest considering adoption: https://onnxscript.ai/intermediate_representation/getting_started.html