Skip to main content
CareerApp

Staff Machine Learning Engineer - AI Foundation

XPeng

Santa Clara, CA · full time · Staff

$215,280 – $364,320

Listed on XPeng’s own careers site. You apply with them directly — we never stand between you and the employer.

What this role is

XPENG is hiring a staff-level ML engineer to optimize large language models and foundation models for autonomous driving applications, focusing on inference efficiency and training acceleration. This role suits someone with deep expertise in transformer optimization, GPU programming, and model deployment who wants to work on cutting-edge AI infrastructure for self-driving vehicles.

Our summary, not XPeng’s wording. The full posting is on their site.

Skills this role names

Log in to see which of these are already on your profile.

What they ask for

Required

  • Master's degree in Computer Science, Computer Engineering, or Electrical Engineering, or equivalent
  • 5-8 years of industry experience
  • Knowledge of PyTorch
  • Understanding of transformer architecture and acceleration techniques

Nice to have

  • Experience in autonomous driving industry
  • Knowledge of TorchScript and NVIDIA TensorRT
  • Strong Python and C++ programming skills
  • Familiarity with GPU, CPU, NPU, and DSP architectures
  • Deep understanding of memory bandwidth and hardware-aware optimization
  • Experience collaborating on complex problems in larger teams

Turn on analytics and we load Google Analytics: Google gets the pages you open and what you do here — searches, jobs you view, jobs you apply to — and sets its own cookies. Leave it off and the only cookies we set are your login, your theme, and this answer. Accept All also records a yes to advertising, which nothing uses yet. Privacy Policy.