Cloud Machine Learning LLM Serving Engineer

Job Area :  Engineering Group, Engineering Group >

Software Engineering

General Summary :

JD for Cloud Machine Learning LLM Serving engineer

Job Overview :

The Qualcomm Cloud Computing team is developing hardware and software for Machine Learning solutions spanning the data center, edge, infrastructure, automotive market. 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 daily so good communication, planning and execution skills are a must.

Key Responsibilities

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.

Implement Kernels for AI workloads

Collaborate and interact with internal teams to analyze and optimize training and inference for deep learning.

Build software tools and ecosystem around AI SW Stack.

Work on vLLM, Triton, ExecuTorch, Inductor, TorchDynamo to build abstraction layers for inference accelerator.

Optimize workloads for both scale-up (multi-SoC) and scale-out (multi-card) systems.

Optimize the entire deep learning pipeline including graph compiler integration.

Apply knowledge of software engineering best practices.

Desirable Skills and Aptitudes

Deep Learning experience or knowledge- LLMs, Natural Language Processing, Vision, Audio, 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.

Experiences to run 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+ years Software Engineering or related work experience.

2+ years experience with Programming Language such as C++, Python.

Minimum Qualifications :

Bachelor's degree in Engineering, Information Systems, Computer Science, or related field.

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 [HIDDEN TEXT] 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  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.

Skills Required

C++, Software Engineering, Deep Learning, Python, Machine Learning, Tensorflow