Q

Qualcomm

Designs mobile chips and licenses wireless patents

AI Performance Engineer Senior Staff - Multiple Teams

Applied Machine LearningConfirmed live in the last 24 hours
Full-timeSan Diego, CA, USAIn-person

About the Job

Requirements

  • Hands-on experience building and optimizing language models, notably with PyTorch and ONNX; production-grade environment experience is preferred.
  • Deep understanding of transformer architectures, attention mechanisms, and performance trade-offs.
  • Experience with workload-mapping strategies involving sharding or other forms of parallelism.
  • Strong Python programming skills.
  • Understanding of computer architecture, machine-learning accelerators, in-memory processing, and distributed systems.
  • Strong communication and problem-solving skills, and ability to learn and work effectively in a fast-paced, collaborative environment.
  • A master's degree in Computer Science, Machine Learning, Computer Engineering, or Electrical Engineering.
  • A bachelor's degree in Computer Science, Engineering, Information Systems, or a related field and 6 or more years of hardware engineering, software engineering, systems engineering, or related work experience; or a master's degree in those fields and 5 or more years of related work experience; or a PhD in those fields and 4 or more years of related work experience.

Responsibilities

  • Convert, optimize, and deploy models for efficient inference using PyTorch and ONNX.
  • Understand advanced generative AI algorithms, including attention mechanisms and mixture-of-experts, and numerics to identify optimization opportunities.
  • Analyze and optimize the inference performance of large language, vision-language, and diffusion models, scaling performance to meet throughput and latency constraints.
  • Map next-generation AI workloads onto current and future hardware designs.
  • Work with customers and internal compiler, firmware, and platform teams to develop solutions.
  • Analyze complex performance or stability issues and identify their underlying root causes.
  • Create engineering solutions that provide continuous insights into AI workload performance and guide ongoing improvements.
  • Design and implement high-level kernels, such as in Triton, with a focus on generating efficient low-level code.

Desired Qualifications

  • Experience building and optimizing language models in production-grade environments.
  • Background in neural-network operators and mathematical operations, including linear algebra and math libraries.
  • Understanding of machine-learning compilers.
  • Experience converging accuracy and evaluating it.
  • Knowledge of torch.compile or torchDynamo.
  • A PhD in Computer Science, Computer Engineering, or Machine Learning.

About the Company

Qualcomm

Designs mobile chips and licenses wireless patents

Qualcomm designs the Snapdragon processors and modems inside many Android phones, and it licenses its wireless patents to device makers around the world. It helped commercialize CDMA, the cellular technology behind 3G networks, and its patents cover core parts of the 4G and 5G standards that phones use to connect. The company does not run its own chip factories; contract foundries make its chips, which it sells to makers of phones, PCs, cars, and connected devices. Device makers also pay Qualcomm royalties for using its patented technology, which forms a second business alongside chip sales. Qualcomm was founded in 1985 and is based in San Diego, California.

Company size
10,001+
Company stage
IPO
Headquarters
San Diego, CA
Founded
1985

Recently Posted Jobs

Explore more opportunities at Qualcomm

See more jobs