A Declarative System for Optimizing AI Workloads