Hardware optimization on Android for inference of AI models