AI-enhanced robots