Sr Staff Engineer - ML Libraries Development Engineer

Qualcomm's CPU architecture team is driving the core math libraries needed for ML/AI acceleration. This position will expose you to Qualcomm's cutting-edge SoC and ML/AI platforms in the industry. You will participate in optimizing core ML kernels using the latest advancements like SME and SVE of the ARM CPU architecture to enhance the performance of ML models on Qualcomm's SoC.

Key Responsibilities:

• Optimize core ML kernels for ARM CPU architecture
• Enhance performance of ML models on Qualcomm SoC
• Work with vector processors and ARM Arch64 ISA
• Design and optimize algorithms for AI/ML applications
• Evaluate and optimize performance for ARM architecture
• Inference ML models written in PyTorch/TensorFlow/Keras

Required Skills:

• Understanding of ARM CPU architecture fundamentals and ARM Arch64 ISA
• Experience in optimizing kernels for vector processors
• Knowledge of basic linear algebra functions used in AI/ML
• Algorithm design skills (logic, critical thinking)
• Performance evaluation and optimization for ARM architecture
• Experience with inferencing ML models in PyTorch/TensorFlow/Keras
• Understanding of open-source library framework design

Preferred Skills:

• Strong programming skills and deep understanding of ARM ISA
• Experience with algorithms suitable for vector and matrix accelerators
• Strong analytical and debugging skills
• Proficiency in optimizing linear algebra algorithms
• Performance evaluation using QEMU, simulators, emulators, and real hardware

This role offers exposure to Qualcomm's advanced SoC and ML/AI platforms, allowing you to work on cutting-edge technology in the industry. You'll be part of a team driving innovation in ML/AI acceleration, optimizing performance, and working with the latest ARM CPU architectures.

Join Qualcomm to unlock your full potential and be part of a team that's shaping the future of ML and AI technologies. Apply today to contribute to world-changing innovations and breakthrough technologies in a supportive, inclusive culture that values diverse expertise and fosters professional growth.