autocast

Modules

modelopt.onnx.autocast.convert(onnx_path[, ...])	Convert model to mixed precision.
modelopt.onnx.autocast.graphsanitizer	Graph sanitization and optimization for ONNX models.
modelopt.onnx.autocast.logging_config	Logging configuration module for AutoCast.
modelopt.onnx.autocast.nodeclassifier	Node classification module for AutoCast.
modelopt.onnx.autocast.precisionconverter	Precision conversion module for ONNX models.
modelopt.onnx.autocast.referencerunner	Reference runner module for ONNX model execution.
modelopt.onnx.autocast.utils	Utility functions for AutoCast.

AutoCast package for converting FP32 ONNX models to mixed precision.