AI workload protection