On-device AI processing