Machine Learning Engineer (Datacentre AI Engineering) - Riyadh, KSA

Company:

Job Area:

General Summary:

About Us

Qualcomm is enabling a world where everyone and everything can be intelligently connected. You interact with products and technologies made possible by Qualcomm every day, including 5G-enabled smartphones that double as pro-level cameras and gaming devices, smarter vehicles and cities, and the technology behind the smart, connected factories that manufactured your latest purchase. Qualcomm 5G and AI innovations are the power behind the connected intelligent edge. You’ll find our technologies behind and inside the innovations that deliver significant value across multiple industries and to billions of people every day.

About the Role

The Qualcomm Cloud Computing team is developing hardware and software for Machine Learning solutions for Datacentre AI. We are seeking ambitious, bright, and innovative engineers with experience in machine learning framework development. Job activities span the whole product life cycle from early design to commercial deployment. The environment is fast-paced and requires cross-functional interaction. Good communication, planning, and execution skills are a must.

Key Responsibilities will include:

• Analyse software requirements, determine the feasibility of design within the given constraints, consult with architecture and HW engineers, and implement software solutions best suited for Qualcomm's SOCs.
• Analyse and identify system level issues, interface with the software development, integration, and test teams
• Lead high-performance teams towards system design and deliverables.
• Proven track record of leading teams in Machine learning software engineering.
• Strong foundation of Mathematical modelling of problems and linear algebra, coupled with state of the art algorithms in ML/AI space.
• Improve and optimize key Deep Learning models on Qualcomm AI 100.
• Build deep learning framework extensions for Qualcomm AI 100 in upstream open-source repositories.
• Collaborate and interact with internal teams to analyse and optimize training and inference for deep learning.
• Build software tools and ecosystem around AI SW Stack.
• Work on Triton, ExecuTorch, Inductor, TorchDynamo to build abstraction layers for inference accelerator.
• Optimise workloads for both scale-up (multi-SoC) and scale-out (multi-card) systems.
• Optimise the entire deep learning pipeline including graph compiler integration.
• Apply knowledge of software engineering best practices.

Desirable Skills and Aptitudes

• Deep Learning experience and knowledge – LLMs, Natural Language Processing, Vision, Audio, Diffusion, Recommendation systems.
• Knowledge of the structure and function of different components of Pytorch, TensorFlow software stacks.
• Excellent C/C++/Python programming and software design skills, including debugging, performance analysis, and test design.
• Ability to work independently, define requirements and scope, and lead your own development effort.
• Well versed with open-source development practices.
• Strong developer with a research mindset – strives to innovate.
• Avid problem solver – should be able to find solutions to key engineering and domain problems.
• Knowledge of tiling and scheduling a Machine learning operator is a plus.
• Experience in using C++ 14 (advanced features)
• Experience of profiling software and optimization techniques
• Hands on experience writing SIMD and/or multi-threaded high-performance code is a plus.
• Experience of ML compiler, Auto-code generation (using MLIR) is a plus.
• Experience in deploying workloads on large scale heterogeneous clusters is a plus.
• Hands-on experience with CUDA, CUDNN is a plus.

Qualifications:

• Bachelor's / Masters/ PHD degree in Engineering, Machine learning/ AI, Information Systems, Computer Science, or related field.
• 2-10 years’ of Software Engineering or related work experience.
• 2-10 years’ experience with Programming Language such as C++, Python.

What's on Offer

Apart from working with great people, we offer the below:

• Salary including housing & transport allowance
• Stock (RSU's) and performance related bonus
• 16 weeks fully paid Maternity Leave
• 6 weeks fully paid Paternity Leave
• Employee stock purchase scheme
• Child Education Allowance
• Relocation and immigration support (if needed)
• Life and Medical Insurance
• Live+ Well Reimbursement for health and recreational membership fees

Minimum Qualifications:

• Bachelor's degree in Engineering, Information Systems, Computer Science, or related field and 2+ years of Software Engineering or related work experience.

• 2+ years of academic or work experience with Programming Language such as C, C++, Java, Python, etc.

• References to a particular number of years experience are for indicative purposes only. Applications from candidates with equivalent experience will be considered, provided that the candidate can demonstrate an ability to fulfill the principal duties of the role and possesses the required competencies.

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 [email protected] 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.

To all Staffing and Recruiting Agencies: Our Careers Site is only for individuals seeking a job at Qualcomm. Staffing and recruiting agencies and individuals being represented by an agency are not authorized to use this site or to submit profiles, applications or resumes, and any such submissions will be considered unsolicited. Qualcomm does not accept unsolicited resumes or applications from agencies. Please do not forward resumes to our jobs alias, Qualcomm employees or any other company location. Qualcomm is not responsible for any fees related to unsolicited resumes/applications.

If you would like more information about this role, please contact Qualcomm Careers.