Staff AI Model Optimization Architect

Qualcomm; San Diego (CA); On-site; USD 158,400 - 237,600; Full time

Benefits offered by this job

Competitive annual discretionary bonus

Annual RSU grants

Highly competitive benefits package

Job summary

Qualcomm is seeking a Staff Engineer – AI Model Optimization Architect in San Diego, California. This position involves leading end-to-end model transformation and optimization for large-scale models on Qualcomm accelerators, collaborating with compiler and performance teams to enhance execution efficiency.

The ideal candidate will have 4+ years of relevant experience, expert-level skills in PyTorch, and a strong background in computer architecture and ML accelerators. The role offers a competitive salary of $158,400 to $237,600 along with a robust benefits package.

Qualifications

• 4+ years of experience in Hardware Engineering, Software Engineering, or related.
• Hands-on experience with torch.compile or related workflows.
• Strong foundation in computer architecture and ML accelerators.

Responsibilities

• Lead end-to-end model optimization for LLMs and multimodal models.
• Architect model optimization strategies for Qualcomm accelerators.
• Debug complex performance issues and drive solutions.

Skills

Expert level expertise in PyTorch

Strong Python engineering skills

Deep understanding of transformer architectures

Knowledge of graph capture and compilation workflows

Education

MS in Computer Science, Machine Learning or related

PhD in relevant field

Tools

Triton

Job description

Qualcomm is seeking a Staff Engineer – AI Model Optimization Architect in San Diego, California. This position involves leading end-to-end model transformation and optimization for large-scale models on Qualcomm accelerators, collaborating with compiler and performance teams to enhance execution efficiency.

The ideal candidate will have 4+ years of relevant experience, expert-level skills in PyTorch, and a strong background in computer architecture and ML accelerators. The role offers a competitive salary of $158,400 to $237,600 along with a robust benefits package.