XPENG

Staff Machine Learning Engineer - Foundation Model

XPENG
Apply
5 months ago
Santa Clara, CA, USAStaff+
H1B sponsor

Base Salary

$215k - $364k/yr

Responsibilities

  • Design and implement large-scale multimodal architectures such as vision-language-action transformers for end-to-end autonomous driving.
  • Develop pretraining and fine-tuning strategies using labeled and unlabeled fleet data including images, video, LiDAR, CAN bus data, maps, and human driving behaviors.
  • Research and integrate cross-modal alignment methods including visual grounding, temporal reasoning, policy distillation, imitation learning, and reinforcement learning.
  • Scale training across thousands of GPUs with distributed training frameworks such as FSDP and DDP.
  • Conduct systematic ablation, evaluation, and visualization of model behavior across perception, reasoning, and planning tasks.
  • Optimize models for onboard deployment, including quantization, export, and latency-accuracy trade-offs.

Requirements

  • Master’s degree or higher in Computer Science, Electrical/Computer Engineering, or a related field.
  • At least 3 years of experience in deep learning research or productization.
  • Strong proficiency in PyTorch and modern transformer-based model design.
  • Experience with large-scale pretraining or multimodal modeling involving vision, language, or planning.
  • Deep understanding of representation learning, temporal modeling, and self-supervised or reinforcement learning techniques.
  • Familiarity with distributed training using DDP or FSDP and large-batch optimization.
  • Preferred: PhD in CS, CE, EE, or a related field with at least 1 year of relevant industry experience.
  • Preferred: Publication record in top-tier AI conferences including CVPR, ICCV, NeurIPS, ICLR, ICML, or ECCV.
  • Preferred: Experience building foundation or end-to-end driving models or LLM/VLM architectures such as ViT, Flamingo, BEVFormer, RT-2, or GRPO-style policies.
  • Preferred: Familiarity with RLHF, DPO, GRPO, trajectory prediction, or policy learning for control tasks.
  • Ability to collaborate cross-functionally with infrastructure, perception, and planning teams to deliver production-ready models.

Benefits

  • Collaborative, research-driven environment with access to massive real-world data and industry-scale compute.
  • Opportunity to work with top-tier researchers and engineers on foundation models for autonomous driving.
  • Snacks, lunches, dinners, and fun activities.
  • Bonus, equity, and benefits are provided in addition to base salary.

Tech Stack

Categories

AI ResearchML Engineering
XPENG

About XPENG

1,001-5,000 employees

XPENG designs, manufactures, and sells smart electric vehicles for consumers, integrating in-house advanced driver-assistance systems and connected in-car software. Founded in 2014 and headquartered in Guangzhou, it operates plants in Zhaoqing and Guangzhou, maintains a European HQ in Amsterdam, and develops eVTOL aircraft via XPENG AEROHT.

Contact me