AI data workload performance optimization