Staff Engineer - ML Libraries Development Engineer

Qualcomm; Bengaluru; On-site; INR 1,800,000 - 2,800,000; Full time

Job summary

Qualcomm India Private Limited in Bengaluru is seeking an ML Libraries Development Engineer with 8 to 12 years of relevant experience. You will work on CPU-based ML acceleration, optimizing core ML kernels and leveraging SME and ARM SVE on Qualcomm's SoCs.

Join a team driving high-performance ML libraries, collaborate across software engineering disciplines, and contribute to industry-leading AI capabilities on mobile/edge platforms.

Qualifications

• Bachelor's degree in Engineering, IT, CS, or related field with 4+ years of software engineering experience.
• Master's degree in Engineering, IT, CS, or related field with 3+ years of software engineering experience.
• PhD in Engineering, IT, CS, or related field with 2+ years of software engineering experience.
• 2+ years of work experience with programming languages such as C, C++, Java, Python.

Responsibilities

• Optimize core ML kernels for ARM CPU using SME and SVE on Qualcomm SoC.
• Enhance performance of ML models running on CPU of the SoC.
• Collaborate on ML libraries development and hardware-aware optimization.

Skills

ARM CPU architecture

Vector processor optimization

Linear algebra for AI/ML

Algorithm design

Performance evaluation

ML model inference (PyTorch/TensorFlow

Open-source library design

Education

Bachelor's degree + 4+ yrs exp

Master's degree + 3+ yrs exp

PhD + 2+ yrs exp

Tools

C/C++/Java/Python

PyTorch/TensorFlow

Job description

Company

Qualcomm India Private Limited

Job Area

Engineering Group, Engineering Group > Software Engineering

General Summary

ML Libraries Development Engineer (8 to 12 years of relevant experience):

=========================================================================

CPU architecture team is driving the core math libraries needed for ML/AI acceleration. This position/s will expose you to Qualcomm’s cutting edge SoC and ML/AI platforms in the industry.

Participate in Optimizing the core ML kernels using the latest advancements like SME, SVE of the ARM CPU architecture and enhance the performance of the ML models on the CPU of the QCOM SoC.

Candidates with 8-12 years of relevant experience & willing to work in Bangalore location are welcome to apply

Required Skills

• Understanding of ARM CPU architecture fundamentals and ARM Arch64 ISA
• Optimizing kernels for vector Processors
• Understanding of the basic linear algebra functions used in AI/ML
• Algorithm design (logic, critical thinking)
• Performance Evaluation and Optimization of the applications for ARM architecture
• Inferencing of the ML models written in Pytorch/TensorFlow/Keras
• Understanding of the typical Open-Source Library framework design

Preferred Skills

• Strong Programming skills and deep understanding of the ARM ISA
• understanding of the algorithms suitable for Vector and matrix accelerators
• Strong Analytical and debugging skills
• Good understanding of Optimizing the Linear Algebra algorithms
• Performance evaluation using QEMU, Simulators, Emulators and on Real Hardware

Minimum Qualifications

• Bachelor's degree in Engineering, Information Systems, Computer Science, or related field and 4+ years of Software Engineering or related work experience.
• Master's degree in Engineering, Information Systems, Computer Science, or related field and 3+ years of Software Engineering or related work experience.
• PhD in Engineering, Information Systems, Computer Science, or related field and 2+ years of Software Engineering or related work experience.
• 2+ years of work experience with Programming Language such as C, C++, Java, Python, etc.

Applicants: Qualcomm is an equal opportunity employer. If you are an individual with a disability and need an accommodation during the application/hiring process, rest assured that Qualcomm is committed to providing an accessible process. You may e-mail [e-mail] or call Qualcomm's toll-free number found here. Upon request, Qualcomm will provide reasonable accommodations to support individuals with disabilities to be able participate in the hiring process. Qualcomm is also committed to making our workplace accessible for individuals with disabilities. (Keep in mind that this email address is used to provide reasonable accommodations for individuals with disabilities. We will not respond here to requests for updates on applications or resume inquiries).

Qualcomm expects its employees to abide by all applicable policies and procedures, including but not limited to security and other requirements regarding protection of Company confidential information and other confidential and/or proprietary information, to the extent those requirements are permissible under applicable law.