AI-based Well Optimization