AI-based flight control