Q

Qualcomm

Designs mobile chips and licenses wireless patents

Data Center LLM Optimization Customer Engineer - Software Applications Engineering

Applied Machine LearningPosted on Nov 9
Full-timeShanghai, ChinaNo salary listedIn-person

About the Job

Requirements

  • A master's degree or higher in Computer Science, Electrical Engineering, Computer Engineering, AI/ML, or a related technical field, with hands-on systems and performance engineering experience.
  • Strong understanding of Transformer and large language model inference, including attention, MHA/GQA/MQA, feed-forward networks, KV cache, prefill versus decode behavior, and memory/computation characteristics.
  • Hands-on experience deploying and profiling large language model workloads using Python and at least one major machine learning framework such as PyTorch; ability to diagnose performance using traces, profiler data, logs, and system telemetry.
  • Experience with model optimization techniques such as quantization, batching, memory optimization, kernel/runtime tuning, and graph/compiler optimization.
  • Understanding of distributed AI concepts, including multi-accelerator execution, collective communication, scale-up/scale-out fabrics, and common parallelism strategies.
  • Strong Linux skills and familiarity with containers and orchestration concepts such as Docker and Kubernetes used in datacenter AI deployments.
  • Strong Python and Bash scripting and automation skills, and a disciplined approach to benchmark reproducibility and performance-data analysis.
  • Debugging and problem-solving skills, and customer-facing communication skills to work across multiple engineering organizations.
  • A bachelor's degree in Engineering, Information Systems, Computer Science, or a related field and 4 or more years of Software Applications Engineering, Software Development, or related work experience; or a master's degree in those fields and 3 or more years of related experience; or a PhD in those fields and 2 or more years of related experience.
  • At least 2 years of experience with a programming language such as C, C++, Java, or Python.
  • At least 1 year of experience with debugging techniques.

Responsibilities

  • Deploy and optimize large language model and multimodal inference workloads on Qualcomm Cloud AI accelerator platforms, from a single accelerator to multi-node and rack-scale configurations.
  • Profile and optimize prefill and decode performance, including Time to First Token, Time per Output Token, inter-token latency, tokens per second, throughput per user, and performance per watt.
  • Analyze model architecture and runtime bottlenecks across compute, accelerator memory, host memory, KV cache, interconnect, networking, and the software stack.
  • Optimize model execution using quantization, batching, continuous/dynamic batching, KV-cache management, paged attention, prompt/context-length tuning, and speculative decoding where applicable.
  • Optimize distributed inference across tensor, pipeline, data, context, and expert parallelism; characterize collective communication including AllReduce, AllGather, ReduceScatter, and AllToAll.
  • Evaluate and tune scale-up and scale-out performance, including topology-aware workload placement, network bandwidth and latency, communication-compute overlap, and scaling efficiency.
  • Design and execute reproducible benchmark and stress-test methodologies across model size, batch size, input/output sequence length, concurrency, accelerator count, and node count; identify saturation points and performance regressions.
  • Bring up and validate customer large language models and representative open-source models; establish performance baselines, optimization plans, and competitive and production-readiness reports.
  • Partner with runtime, compiler, firmware, driver, platform, networking, and model teams to debug cross-layer performance issues and drive root-cause resolution.
  • Build Python- and Bash-based profiling, benchmarking, automation, and data-analysis tooling; document reusable optimization recipes, reference configurations, and best practices.
  • Work directly with customers during evaluation, proof-of-concept, design-win, and production phases; convert workload requirements into engineering actions and communicate performance trade-offs to technical and business stakeholders.

Desired Qualifications

  • Experience with high-performance large language model serving stacks or optimization frameworks such as vLLM, TensorRT-LLM, SGLang, Hugging Face Transformers, DeepSpeed, or equivalent accelerator-specific runtimes.
  • Experience with model families such as Llama, Qwen, DeepSeek, Mixtral, other dense or mixture-of-experts large language models, and multimodal models.
  • Knowledge of mixture-of-experts architectures, expert parallelism, token routing, and AllToAll communication performance.
  • Experience with RDMA, RoCE, InfiniBand, or Ethernet-based AI fabrics, NUMA/topology awareness, and distributed performance analysis.
  • Familiarity with MLPerf Inference methodology, production observability, CI/CD performance regression testing, and capacity planning.
  • Experience supporting customer proof-of-concepts or production deployments of AI accelerators at server, rack, or cluster scale.

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