AI for IoT devices