Staff Machine Learning Engineer - Foundation Model
$215.28k - $364.32kFull-time
XPENG
XPENG is a leading smart technology company at the forefront of innovation, integrating advanced AI and autonomous driving technologies into its vehicles, including electric vehicles (EVs), electric vertical take-off and landing (eVTOL) aircraft, and robotics. With a strong focus on intelligent mobility, XPENG is dedicated to reshaping the future of transportation through cutting-edge R&D in AI, machine learning, and smart connectivity. We are looking for a full-time Machine Learning Engineer / Research Scientist to drive the modeling and algorithmic development of XPENG’s next-generation Vision-Language-Action (VLA) Foundation Model — the core brain that powers our end-to-end autonomous driving systems. You will work closely with world-class researchers, perception and planning engineers, and infrastructure experts to design, train, and deploy large-scale multi-modal models that unify vision, language, and control. Your work will directly shape the intelligence that enables XPENG’s future L3/L4 autonomous driving products. Key Responsibilities
- Design and implement large-scale multi-modal architectures (e.g., vision–language–action transformers) for end-to-end autonomous driving.
- Develop pretraining and fine-tuning strategies leveraging massive labeled and unlabeled fleet data (images, video, LiDAR, CAN bus, maps, human driving behaviors, etc.).
- Research and integrate cross-modal alignment (e.g., visual grounding, temporal reasoning, policy distillation, imitation and reinforcement learning) to improve model interpretability and action quality.
- Collaborate with infrastructure engineers to scale training across thousands of GPUs using distributed training frameworks (FSDP, DDP, etc.).
- Conduct systematic ablation, evaluation, and visualization of model behavior across perception, reasoning, and planning tasks.
- Contribute to model deployment optimization, including quantization, export, and latency–accuracy trade-offs for onboard execution.
- Master’s degree or higher in Computer Science, Electrical/Computer Engineering, or related field, with 3+ years of experience in deep learning research or productization.
- Strong proficiency in PyTorch and modern transformer-based model design.
- Experience in large-scale pretraining or multi-modal modeling (vision, language, or planning).
- Deep understanding of representation learning, temporal modeling, and self-supervised or reinforcement learning techniques.
- Familiarity with distributed training (DDP, FSDP) and large-batch optimization.
- PhD in CS/CE/EE or related field, with 1+ years of relevant industry experience.
- Publication record in top-tier AI conferences (CVPR, ICCV, NeurIPS, ICLR, ICML, ECCV).
- Prior experience building foundation or end-to-end driving models, or LLM/VLM architectures (e.g., ViT, Flamingo, BEVFormer, RT-2, or GRPO-style policies).
- Familiarity with RLHF/DPO/GRPO, trajectory prediction, or policy learning for control tasks.
- Proven ability to collaborate cross-functionally with infra, perception, and planning teams to deliver production-ready models.
- A collaborative, research-driven environment with access to massive real-world data and industry-scale compute.
- An opportunity to work with top-tier researchers and engineers advancing the frontier of foundation models for autonomous driving.
- Direct impact on the next generation of intelligent mobility systems.
- Opportunity to make significant impact on the transportation revolution by the means of advancing autonomous driving.
- Competitive compensation package.
- Snacks, lunches, dinners, and fun activities.
Vacancy posted 22 days ago
Similar jobs that could be interesting for youBased on the Staff Machine Learning Engineer - Foundation Model in California vacancy
- ...Description About the Institute of Foundation Models We are a dedicated... ..., data scientists, and engineers, tackling the most fundamental... ...performance computing in deep learning, driving impactful discoveries... ...foundation models to unlock machine intelligence beyond lingual....FoundationVisa sponsorship
Institute of Foundation Models
Sunnyvale, CA24 days ago $174.72k - $295.68k
...through cutting-edge R&D in AI, machine learning, and smart connectivity. We are seeking Machine Learning Engineers with strong expertise in generative modeling and large-scale deep learning systems... ...Experience with multimodal foundation models and video tokenizers or VAEs...FoundationFull timeXPENG Motors
Santa Clara, CA24 days ago$175k - $296k
...experience of the future. We are looking for a full-time Machine Learning Engineer, with deep knowledge and strong enthusiasm towards... ...state-of-art ML infrastructure for training very large foundation model and accelerating model training/inference. Our mission...FoundationFull time- Stanford University is seeking a Machine Learning Engineer to support advanced technical research within... ..., and clinical utility of ML/AI models intended for translational impact and... ...closely with senior faculty and research staff, the individual will build end-to-end...Foundation
Stanford University
Stanford, CA13 days ago $298k - $368k
...builds the system which learns the spatial-temporal... ...of sensors, enabling engineers like you to (1) develop... ...data, to (2) develop models and model training at... ...years of experience in Machine Learning, with a focus... ...(LLM, VLM, or similar foundation models). ~ Proven expertise...FoundationFull timeRemote work$117.7k - $221.4k
...sits at the intersection of machine learning, data infrastructure, and developer... ...not only on stronger models, but also on better infrastructure... ...model reflects how Cola engineers think: build durable... ...featurization, and inference foundations that power scalable world understanding...FoundationFull timeLocal areaRemote workWork from homeRelocation packageFlexible hoursGeneral Motors
Sunnyvale, CA2 days ago$189.3k - $290.7k
...and we are looking for a Staff AI/ML Software Engineer to lead the adaptation, fine... ..., and distillation of foundation models for the automotive edge.... ...curation for complex human-machine interaction use cases,... ...and own the reinforcement learning flywheel, implementing human...FoundationFull timeWork at officeLocal areaWork from homeRelocation package3 days per weekGeneral Motors
Mountain View, CA5 days ago- ...of patients worldwide.We’re a team of engineers, clinicians, and innovators united by... ...Engineer, you will develop and fine-tune the foundation models—VFMs, VLMs, and VLA models—that let... ...that enable rapid prototyping and learning while building toward a product solution...FoundationLocal areaWorldwideFlexible hours
Intuitive Surgical
Sunnyvale, CAa month ago $131k - $180k
...global leader in materials engineering solutions used to... ...use every day. As the foundation of the global electronics... ...encourages you to learn, develop, and grow your... ...applying cutting-edge AI and machine learning to accelerate... ...AI and large language models to tackle complex...FoundationFull timeRelocationAPPLIED MATERIALS
Santa Clara, CA16 hours ago$298k - $368k
...the system which learns the spatial-temporal... ...sensors, enabling engineers like you to (1)... ...data, to (2) develop models and model training... ...to a Senior Staff Technical Lead Manager... ...teams to optimize foundational architectures for... ...Science, Robotics, Machine Learning, or equivalent...FoundationFull timeRemote workShift work- ...intelligence to every moving machine on the planet. Applied... ...Bangalore; Seoul; and Tokyo. Learn more at applied.co. We are... ...on pretraining world-action foundation model with various world modalities... ...the Research Scientists and Engineers on high quality research publications...FoundationFor contractorsFor subcontractorCasual workInternshipWork at officeImmediate startRemote workDay shift
Applied Intution
Sunnyvale, CA2 days ago $40 per hour
...As a Machine Learning Engineering Intern, you will be part of a collaborative team supporting the development... ..., and monitoring of machine learning models. Your primary role is to contribute... ...sustainability on a single, secure foundation—ensuring data is trusted, traceable,...FoundationRemote jobFull timeSummer workInternshipSummer internshipWork at officeWorkiva
Sacramento, CA4 days ago- ...training frameworks for XPENG physical AI foundation models, including VLA 2.0, XWorld, and... ...closed-loop simulation, reinforcement learning, and enterprise LLM/VLM applications.... ...degree in computer science, computer engineering, or electrical engineering, or equivalent...FoundationFull time
XPENG
Santa Clara, CA3 days ago $200k - $300k
...the top of their industry is seeking a Machine Learning Engineer to join its research and development... .... Rather than maintaining existing models, this individual will help invent entirely... ...Strong mathematical and statistical foundations Bachelor's degree in Computer Science...FoundationFull timeWorldwideMotion Recruitment
San Francisco, CA14 days ago$174.72k - $295.68k
...through cutting-edge R&D in AI, machine learning, and smart connectivity.Our mission is to build strong foundation for LLM deployment and... ...and is not limited to: LLM model fine tuning, PTQ, QAT, on-vehicle... ...programming and software engineering skills.Ability to work effectively...FoundationFull timeXPENG Motors
Santa Clara, CA23 days ago$180k - $220k
...build sensors and tools for engineers, roboticists, and researchers... ...for a highly technical Machine Learning Engineer to lead our efforts... ...simply be "importing" pre-made models; you will be architecting deep... ...-Shot Learning.Mathematical Foundation: Strong grasp of linear...FoundationWork experience placementLocal areaOuster
San Francisco, CAa month ago- ...Description About the Institute of Foundation Models We are a dedicated... ..., data scientists, and engineers, tackling the most fundamental... ...performance computing in deep learning, driving impactful discoveries... ...performance for the machine learning software stacks, especially...FoundationWork experience placementVisa sponsorship
Institute of Foundation Models
Sunnyvale, CA24 days ago $195k - $230k
...the RoleWe are looking for a Senior Machine Learning Engineer to help evolve our large-scale recommendation... ...issues related to data quality, model drift, and system performance in... ....AI & LLM ApplicationsApply LLMs and foundation models to recommendation-related problems...FoundationFull timeLocal areaWork from homeNews Break
Mountain View, CA29 days ago$230k - $265k
...alongside industry-veteran scientists and engineers. As a Senior Machine Learning Engineer, you’ll bring your strong... ...for large language and speech models using PyTorch and/or JAX, making... ...and fine-tuning large language or foundation models, with production experience...FoundationPermanent employmentOtter.ai
Mountain View, CAa month ago$61k - $101k
...with training and work experience in Machine Learning, LLM/NLP, or a related field. We need... .... We look for experience in foundation model training, including pre-training from... ...based formats. We need solid software engineering capabilities and the ability to build...FoundationFull timeWork experience placementJ.P. Morgan
Palo Alto, CA14 days ago$181k - $250k
...ownership, continuous learning, and thoughtful speed.... ...build out the evaluation foundation across all of these... ...You'll work on voice models, agent capability evals... ...in Computer Science, Machine Learning, Statistics,... ...years of experience in ML Engineering or Applied ML with 8+...FoundationContract workWorldwideAircall.io, Inc.
San Francisco, CA5 days ago$168.1k - $312.3k
...data and novel computational models to drive impact. Seamless data... ...is to develop AI models that learn from biological data at... ...motivated and passionate Senior ML Engineer to join our Generative... ...team and help build and scale foundation models and agentic systems for...FoundationFull timeLocal areaWorldwideRelocation packageGenentech
South San Francisco, CA25 days ago$61k - $101k
...training and hands-on work experience in Machine Learning, LLM/NLP, or a related field. We... .... We require experience training foundation models from scratch, including token and... ...formats. We need strong software engineering skills and the ability to build robust...FoundationFull timeWork experience placementJ.P. Morgan
Palo Alto, CA7 days ago$283.5k - $381.6k
...Our advanced AI software and foundation models enable vehicles to perceive,... ...of excellence, constantly learning and evolving as we pave the... ...career! The Role As an ML Engineer within the Application... ...set you up for success as a Machine Learning Engineer at Wayve,...FoundationFull timeWork from homeWayve
California25 days ago- ...Description About the Institute of Foundation Models We are a dedicated... ..., data scientists, and engineers, tackling the most fundamental... ...performance computing in deep learning, driving impactful discoveries... .... The Role As a Machine Learning Engineer at the Institute...FoundationWorldwideVisa sponsorship
Institute of Foundation Models
Sunnyvale, CA24 days ago $259.03k - $311.43k
...to explore, create, play, learn, and connect with friends in... ...we are seeking experienced machine learning engineers who thrive on solving complex... ...deep learning-based models for performance advertising... ...business. Your work will lay the foundation to deliver effective...FoundationFull timeWork experience placementH1bWork at officeLocal areaVisa sponsorshipMonday to FridayRoblox
San Mateo, CAa month ago$200.35k - $260.91k
...The OpportunityWe’re hiring a Senior Machine Learning Engineer to join our AI team, reporting directly... ..., why now, and who is deciding. Your models power products customers use every... ...with clear end-to-end ownership.Strong foundation in machine learning and applied...FoundationFull time6sense
San Francisco, CA14 days ago- ...AI Agents). Realm-X serves as both a foundation for internal teams to build and scale... ...Are Looking ForWe're hiring a Senior Machine Learning Engineer to design and ship the next generation... ...across frontier LLMs, smaller models, and routing strategies.Lead a Pod: Lead...FoundationFull timeFlexible hours
AppFolio
Santa Barbara, CA24 days ago - ...architectures, combining rigorous engineering with learning systems proven in globally... ...and implement scalable machine learning pipelines for... ..., and deploy deep learning models using PyTorch, TensorFlow,... ...MLflow, or Kubeflow. Strong foundation in geometric computer vision...FoundationLocal area
FieldAI
Irvine, CA24 days ago $244k - $320k
...our AI-powered personalization engine delivers bespoke experiences... ...and the Human Rights Campaign Foundation's Corporate Equality Index!About the RoleOur Machine Learning Engineering team powers personalized... ..., from data exploration and modeling to deployment, monitoring, and...FoundationFull timeAttentive
San Francisco, CAa month ago
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Staff Machine Learning Engineer - Foundation Model. Be the first to apply!
Related searches
- engineering aide California
- technology administrator California
- senior staff systems engineer California
- senior staff engineer California
- staff engineer California
- assistant engineer California
- assistant electrical engineer California
- software engineer staff California
- machine learning engineer California
- senior ml engineer California