Data platform for AI workloads