ML inference optimization