Principal Edge AI Software Engineer - Embedded AI ML

Azimuth AI 

📍 Beijing, Beijing, China, China 🇨🇳

full-time
senior
Posted —

Key Skills

AIMLTensorFlowPyTorchquantization

Industry

SemiconductorConsumer Electronics

Job Description

该职位来源于猎聘 We are seeking an experienced Edge AI Software Architect to lead the design and implementation of advanced machine learning solutions for edge devices and embedded systems. This role focuses on deploying and optimizing large language models (LLMs) and other AI models on resource-constrained hardware.

Responsibilities: Architecture & Design Design and architect scalable Edge AI inference engines for microcontrollers, edge devices, and embedded systems Define technical roadmaps for deploying LLMs and foundation models on edge hardware Lead the architecture of model compression, quantization, and optimization pipelines for resource-constrained devices LLM & Large Model Optimization Optimize and deploy Large Language Models (LLMs) on edge devices using techniques such as quantization (INT8, INT4), pruning, and knowledge distillation Implement model compression techniques to reduce model size while maintaining accuracy for edge deployment Design and optimize inference pipelines for transformer-based models and other foundation models on low-power devices Develop custom kernels and operators optimized for edge AI accelerators Model Development & Deployment Train, fine-tune, and optimize machine learning models using TensorFlow, PyTorch, and ONNX for edge deployment Implement model conversion workflows (TensorFlow Lite, ONNX Runtime, TensorRT, OpenVINO) for various edge platforms Design and implement efficient model serving architectures for edge devices with latency and power constraints Performance Optimization Optimize ML algorithms and inference engines to meet strict performance, power, and memory constraints Profile and optimize model performance on various edge AI accelerators (NPU, DSP, GPU) Achieve low-latency, high-throughput inference while minimizing power consumption Requirements: Education Master's or Ph.D. degree in Computer Science, Electrical Engineering, Machine Learning, or related field Bachelor's degree with 8+ years of relevant experience may be considered Technical Skills Deep expertise in LLM optimization and deployment: quantization, pruning, distillation, LoRA, QLoRA Strong proficiency in ML frameworks: TensorFlow, PyTorch, ONNX, TensorFlow Lite, PyTorch Mobile Expert-level programming skills in C/C++ and Python Extensive experience in embedded software development and real-time systems Proven track record of deploying ML models (especially LLMs) to production edge devices Strong understanding of computer architecture, memory hierarchies, and hardware acceleration Professional Experience 5+ years of experience in embedded ML or Edge AI development Demonstrated experience optimizing and deploying large models (>1B parameters) on edge devices Proven ability to architect and deliver complex ML systems from concept to production Experience with model compression achieving >10x size reduction with minimal accuracy loss Soft Skills Excellent ability to read, understand, and implement research papers in English Strong problem-solving skills and architectural thinking Outstanding communication skills for technical documentation and cross-team collaboration in a global working environment Experience with multimodal models (vision-language, audio-text) on edge devices Contributions to ML optimization frameworks or edge inference engines Understanding of security considerations for edge AI deployment