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 26 days 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$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 workWaymo
San Francisco, CAmore than 2 months 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 time$150k
...Machine Learning Engineer About the Institute of Foundation Models: We are a dedicated research lab for building, understanding, using, and risk-managing foundation models. Our mandate is to advance research, nurture the next generation of AI builders, and drive transformative...FoundationVisa sponsorshipInstitute of Foundation Models
Sunnyvale, CA2 days 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 workWaymo
San Francisco, CA22 days ago- 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
$265k - $331k
...generations. Role Summary We are seeking a Sr. Staff Machine Learning Engineer to lead model pretraining efforts for our Large Driving Model, an... ...data. This role is focused on building the foundation model layer for self-driving: scaling larger models,...FoundationFull timeContract workTemporary workPart timeLocal areaShift 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 hours$170k - $216k
...team builds the system which learns the spatial-temporal representation... ...set of sensors, enabling engineers like you to (1) develop... ...real-world data, to (2) develop models and model training at scale,... ...experience ~3+ years experience in Machine Learning and/or Computer...Full timeRemote workWaymo
San Francisco, CAmore than 2 months 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
- ...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... ...robust data and systems are the foundation of any production...FoundationFull timeSeasonal work
Roblox
San Mateo, CAmore than 2 months ago $213k - $263k
...collaborative group of software engineers, machine learning (ML) engineers, and data... ...machine learning, we model the real world, including realistic... ...technologies such as large foundational World Model trained on our... ...will report to a Senior Staff Tech Lead Manager You...FoundationFull timeRemote workWaymo
Mountain View, CAmore than 2 months ago- ...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 time
Roblox
San Mateo, CAmore than 2 months ago - ...is an applied research lab curating data solutions for foundation model development. We serve every frontier AI lab with the... ..., Meta Superintelligence Labs, and Microsoft AI. Machine Learning Engineer, Quality Intelligence Overview AfterQuery builds...FoundationFull time
AfterQuery
San Francisco, CAmore than 2 months ago $225k - $300k
...Machine Learning Engineer About Latent Health Healthcare today is only... ...longitudinal patient history) Our models are designed to answer... ...and post-training of foundation models Novel objectives... ...primarily hiring for senior and staff-level engineers who are comfortable...FoundationFull timeWork at officeImmediate startLatent
San Francisco, CAmore than 2 months ago$204k - $259k
...In this role, you will: Design, train, and deploy machine learning models to automate the creation of Waymo's HD maps, unlocking scale... ...with partner ML teams, such as Waymo Perception and Waymo AI Foundations, to adapt cutting-edge research into scalable, reliable,...FoundationFull timeRemote workWaymo
Mountain View, CAmore than 2 months ago- ...directly on our Glassdoor page! Machine Learning Engineer @ Clay Clay's ambition is to build... ...learning including data lake foundations and serving infrastructure. Evaluate... ...ML-heavy software engineering, with models and ML-powered features shipped to production...FoundationFull time
Clay Labs
San Francisco, CAa month ago $213k - $263k
...and innate world knowledge of massive foundation models directly onto the Waymo Driver. We... ...Drive cross-functional collaboration to engineer robust, high-reliability training... ...: ~ BS or MS in Computer Vision, Machine Learning, Robotics, or a related field. ~4+...FoundationFull timeRemote workWaymo
Mountain View, CAmore than 2 months ago$90 per hour
...and Jack Dorsey . Position: Machine Learning Engineer Expert Type: Contract... ...solutions for challenging prediction and modeling problems. Analyze datasets and define... ...optimization, transfer learning, foundation model fine-tuning, or reinforcement learning...FoundationFull timeContract workSummer workRemote workMercor
San Francisco, CAmore than 2 months ago- ...company We’re a team of engineers, neuroscientists, and... ...and pioneering models decode emotion, putting... ...that enables us to build foundation models of emotion.... ...demand intense, fast-paced learning. You will: Critically... ...implement the best machine learning approaches...FoundationFull time
Orbit Neuro Co.
San Francisco, CAmore than 2 months ago $195k - $230k
...Role We 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... ...Applications Apply LLMs and foundation models to recommendation-related problems...FoundationFull timeLocal areaWork from homeNews Break
Mountain View, CAmore than 2 months ago$165k - $185k
...offices to seamlessly discover, model, and manage their private... ...to add an AI / Extraction Engineer to the team! There's a lot for... ...help us lay the technical foundation for the company's future.... ...Build, train, and improve machine-learning models for production use....FoundationRemote jobFull timeAllocate
Palo Alto, CAa month ago- ...collection. In order to train a reliable AI model, you need access to the right... ...ICML, AAAI. Role Description As a Machine Learning Engineer at Advex, you will play a pivotal... ...the architectural roadmap, laying the foundation for the Advex platform. Your technical...FoundationFull time
Openreq
San Francisco, CAa month ago $165k - $230k
...role We're looking for exceptional Machine Learning Engineers focused on Ads to help take... ...and advertising systems—building the models and infrastructure that determine how... ...evaluation, and other techniques for adapting foundation models to specific creative and advertising...FoundationFull timeWork at officeRemote workWorldwide3 days per weekHiggsfield
San Francisco, CAa month ago$175k - $275k
...curated datasets, or full-cycle data engineering, Abaka AI provides the foundation for building high-performance AI... ...We’re hiring our first Machine Learning Engineer in the United States, a... ...drive the technical roadmap for model development across multiple modalities...FoundationFull timeImmediate startFlexible hoursAbaka Ai
Palo Alto, CAmore than 2 months ago$204k - $259k
...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... ...for cutting-edge VLM foundation models. Conduct research... ...of experience in Machine Learning, with a focus...FoundationFull timeRemote workWaymo
San Francisco, CAmore than 2 months ago$242k - $290k
...optimize high-performance deep learning models that generate dense,... ...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 packageZoox
Foster, CAmore than 2 months ago- ...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...FoundationFull timeCurrently hiringWork at officeRemote workRelocationRelocation package
Pinterest
San Francisco, CAmore than 2 months ago - ...healthcare, this is the place. The Role As a Senior Machine Learning Engineer, you will build the intelligence layer that automates... ...systems role for someone who knows how to turn modern foundation models into dependable production infrastructure. You should be...FoundationFull timeWork at office
Hike Medical
San Francisco, CAmore than 2 months ago $160k - $215k
...amassing a world class team of engineers, scientists, team builders... ...down-to- earth. The Machine Learning Engineer will work in close... ...including data preprocessing, model training, evaluation, and... ...Understanding of mathematical foundations relevant to algorithm development...FoundationFull timeRemote workCellanome
San Diego, CA15 days 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
- senior staff engineer California
- senior staff systems engineer California
- engineering aide California
- software engineer staff California
- assistant engineer California
- technology administrator California
- staff engineer California
- computer vision machine learning engineer California
- senior ml engineer California
- machine learning engineer California