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 1 day ago
Similar jobs that could be interesting for youBased on the Staff Machine Learning Engineer - Foundation Model in California vacancy
$174.72k - $295.68k
...through cutting-edge R&D in AI, machine learning, and smart connectivity.We are... ...for a full-time Machine Learning Engineer / Research Scientist to drive the modeling and algorithmic development of... ...Vision-Language-Action (VLA) Foundation Model — the core brain that powers...FoundationFull time$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 time- ...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, CA3 days ago - Apple Inc. seeks a senior hands-on Machine Learning Engineering Manager to lead a small team focused on evaluating foundation models and agentic systems, diagnosing failure modes, and driving automated improvements to prompts, contexts, tools, and data generation. You will...Foundation
$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 hours$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- ...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
$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 timeWorldwide$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 area$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 home$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 time$196k - $221k
...alongside industry-veteran scientists and engineers. As a Machine Learning Engineer, you’ll bring your strong... ...for large language and speech models using PyTorch and/or JAX, making principled... ...large language models or other foundation models, including an understanding of...FoundationPermanent employment$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 package$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 Friday$254k - $350k
...optimize high-performance deep learning models that generate dense,... ...on sparse sensor modalities .Engineer temporal processing modules... ...Computer Science, Robotics, Machine Learning, or related field with... ...Language Model, or multi-modal 3D foundation model, or World Model, or...FoundationFull timeTemporary workRelocation package$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 time$196.75k - $243.29k
...to explore, create, play, learn, and connect with friends in... ...are hiring our first Senior Machine Learning engineer within our team. As a... ...production.Build time-series models to predict capacity exhaustion... ...robust data and systems are the foundation of any production...FoundationFull timeWork experience placementSeasonal workH1bWork at officeLocal areaVisa sponsorshipMonday to Friday$184k - $287.5k
Intelligent machines powered by Artificial Intelligence computers that can learn, reason and interact with people... ...has provided the foundation for machines to learn... ...extraordinary Senior Perception Engineer to develop and... ...deep learning models for lane graph construction...FoundationOdd jobFull timeWork experience placementRemote work$150k - $180k
...Systems group where you will be developing machine learning solutions that enhance performance in... ...with UX designers, infrastructure engineers, and other research scientists to develop... ...or electronic warfareStrong foundation in supervised and unsupervised learning...FoundationFull timeContract workWork at officeLocal area$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 time$141.9k - $190.3k
Job Posting Title:Sr Machine Learning EngineerReq ID:10156068Job Description... ...is a global organization of engineers, product developers,... ...to advance the technological foundation and consumer media touch points... ...will help build services and models enabling efficient ad fills...FoundationFull timeWork experience placementWork at office$161.26k - $332.01k
...abilities, we’ll explore your foundational skills and how you... ...representation learning, heterogeneous graph neural... ...networks, generative modeling, recommender systems,... ...is a small group (~6 engineers along with a product prototyping... ....M.S. or PhD in Machine Learning, Computer...FoundationCurrently hiringWork at officeLocal areaRemote workRelocationRelocation package$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 time$150k - $250k
...growing teams.We are seeking a highly skilled Machine Learning Engineer with deep expertise in developing Bird’s Eye View (BEV) fusion models using multimodal sensor inputs,... ...integration in perception pipelines.Solid foundation in computer vision, deep learning, and 3...FoundationFull time- ...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
- ...LLMs, our proprietary models, and a sophisticated Agentic... ..., and continuously learn and adapt.Moveworks is... ...with Moveworks’ Reasoning Engine and natural language... ...software engineer with machine learning expertise to join... ...class LLMs, multimodal foundation models, and hybrid...FoundationWork at officeImmediate startRemote workFlexible hours
$158.4k - $217.8k
...sustainability, design and engineering, ambition and integrity. In... ...future of mobility.Senior Machine Learning Engineer, End-to-End Planning... ...involving vision-action models, vision-language-action (VLA... ...autonomous driving.Strong foundation in physics, probability, and...Foundation$173k - $259k
...express themselves, live in the moment, learn about the world, and have fun... ...full stack of generative AI, including foundational models, efficient infrastructure, and on-device... ...like Spectacles.We're looking for a Machine Learning Engineer to join our Generative ML team!What...FoundationFull timeLive inWork at officeLocal areaWorldwide$160k - $240k
...computer vision, and large language models (LLMs), along with in-house model training... ...results-focused, and deeply skilled Machine Learning Engineers/scientists to work with us on a range... ...outputs.Other QualificationsAcademic Foundation: Advanced degree (Master’s or Ph.D.)...FoundationFull timeWork at officeRemote workHome officeFlexible hours$112k - $269k
SummaryYelp engineering culture is driven by our values... ...use of cutting-edge Machine Learning (ML) and Artificial Intelligence... ...locations. As a Staff-level ML Engineer on... ...NNs), large language models (LLMs), and various... ...and our product. The foundation of our diversity...FoundationWork experience placementLocal areaRemote work
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
- staff engineer California
- assistant engineer California
- engineering aide California
- senior staff engineer California
- senior staff systems engineer California
- software engineer staff California
- technology administrator California
- computer vision machine learning engineer California
- machine learning engineer California
- senior ml engineer California