1 day ago
Shanghai, China or Beijing, ChinaSenior
Responsibilities
- Develop and optimize AI kernels and operators for Qualcomm Cloud AI platforms.
- Enable and optimize LLM, MoE, and multimodal models for large-scale cloud inference.
- Improve accuracy, performance, resource utilization, and memory footprint across the software stack.
- Optimize graphs, compilers, kernels, ROM/RAM efficiency, operator fusion, scheduling, and code generation.
- Analyze performance bottlenecks using profiling, benchmarking, and modeling methodologies.
- Collaborate with performance and modeling teams to deliver efficient inference solutions.
Requirements
- Master's degree or above in Computer Science, Artificial Intelligence, Mathematics, Electrical Engineering, or a related field.
- 5+ years of experience in cloud AI inference, kernel/operator development, or AI compiler technologies.
- Proficiency in C++, Python, and Linux development environments.
- Deep understanding of Transformer architectures, including LLMs, MoE, and modern foundation models.
- Strong experience with MLIR, Triton, TVM, or similar compiler frameworks, including graph transformation, operator lowering, scheduling, operator fusion, kernel auto-tuning, code generation, and memory scheduling.
- Strong experience with parallel computing architectures, vectorization, tiling, memory hierarchy optimization, workload scheduling, graph optimization, model accuracy validation, performance optimization, and memory optimization.
- Strong English verbal and written communication skills.
- Preferred: experience optimizing Attention, FlashAttention, KV Cache, Embedding, GEMM, Softmax, and MoE routing operators.
- Preferred: hands-on development of custom operators and high-performance kernels for GPUs, NPUs, TPUs, or custom AI hardware.
- Minimum qualifications also accept a bachelor's degree with 2+ years of related experience, a master's degree with 1+ year of related experience, or a PhD, plus 2+ years of academic or work experience with programming languages such as C, C++, Java, or Python.
Categories
About Qualcomm
Qualcomm is a public semiconductor company headquartered in San Diego, founded in 1985, that designs and sells wireless chipsets and platforms for mobile devices, automotive, IoT, and networking, notably the Snapdragon application processors and 5G modems. It also licenses a large portfolio of cellular patents to device makers, generating revenue alongside chip sales; its technology underpins many Android smartphones and emerging automotive and edge-compute systems, and it trades on NASDAQ as QCOM.