Inference speed improvement