AI-driven control systems for heavy machinery