AI workload optimization