Latest AI JobsBrowse Jobs
wayve logo

Principal Machine Learning Engineer, 3D Geometric Vision

wayveSunnyvale, California USAAV Engineering
Apply on company site
Salary
$370,040 – $419,800
Job type
Full-time
Work mode
Remote
Experience
Senior
Posted
5 Oct 2026
Status
Open · checked 7d ago

About this role

Based in Sunnyvale, California USA, Wayve is hiring a Principal Machine Learning Engineer for 3D Geometric Vision to design and scale foundation models for autonomous vehicles. This hands-on technical leadership position suits an experienced machine learning professional who excels at taking complex 3D vision research projects through to real-world vehicle deployment.

Role Overview

As a Principal Machine Learning Engineer, the focus centers on building geometric vision and 3D foundation models within the Model Foundations team. The position involves setting the technical direction for geometric vision, prototyping new architectures, building data pipelines, and integrating models into production autonomous driving systems.

Key Responsibilities

Design and train 3D foundation models and world models using large-scale driving data.
Develop model architectures for 3D perception, geometric reasoning, reconstruction, and world modeling across space and time.
Build scalable data generation and auto-labeling pipelines that produce high-quality geometric supervision from large volumes of sensor data.
Develop and scale offline SLAM and 3D reconstruction systems and pipelines to recover accurate trajectories, scene geometry, and calibration signals.
Apply techniques in multi-view geometry, neural rendering, NeRFs, Gaussian Splatting, implicit 3D representations, and feedforward 3D modeling.
Train and evaluate models at scale on distributed compute while rapidly iterating on architectures, objectives, and training recipes.
Optimize and deploy models into production autonomous driving systems under onboard constraints.

Requirements

Deep expertise in 3D computer vision, geometric vision, or 3D machine learning.
Strong experience designing, training, and evaluating modern deep learning models at scale using PyTorch or a comparable framework.
Strong mathematical foundations in geometry, linear algebra, probability, optimization, and 3D transformations.
Excellent software engineering skills in Python and C++.
Proven track record of taking difficult research problems from idea to working system.
Principal-level technical leadership capability to set research and engineering direction and make strong architectural decisions.

Nice to Have

Familiarity with dense 3D reconstruction, neural fields, NeRFs, Gaussian Splatting, or feedforward 3D models.
Experience with large-scale vision pre-training, self-supervised learning, video models, or generative models.
Knowledge of offline SLAM, structure-from-motion, calibration, auto-labeling, and ground-truth generation.
Background in multimodal perception and fusion across camera, radar, and LiDAR.
Experience with distributed training and deploying neural networks on real-time, resource-constrained hardware.

Skills and Tools

PyTorch
Python
C++
NeRFs
Gaussian Splatting
SLAM
LiDAR
Radar

Salary and Benefits

The salary range for this role is $370,040 to $419,800. Additional benefits include annual salary benchmarking, meaningful equity, relocation support and visa sponsorship where applicable, a hybrid working model, learning and development budgets, and comprehensive health, dental, and wellness benefits.

About the Company

Wayve builds embodied intelligence designed to move real vehicles safely through complex urban environments. The company utilizes an end-to-end AI approach that allows vehicles to learn directly from real-world experience without relying on hand-coded rules or pre-mapped environments. Operating across multiple global hubs, the organization partners with global OEMs to advance automation from level 2 plus up to level 4.

How to Apply

Candidates should apply directly on the official company careers page.

Skills and tech stack

PyTorchPythonC++NeRFsGaussian SplattingSLAMLiDARRadarComputer VisionDeep LearningMachine Learning

Questions candidates ask

What is the salary range for this role?

The salary range is between $370,040 and $419,800.

Where is the job located?

The position is located in Sunnyvale, California USA, with a hybrid working model.

What technical frameworks and languages are required?

The role requires software engineering skills in Python and C++, along with experience using PyTorch or a comparable deep learning framework.

What does the interview process involve?

The process includes an initial recruiter screen, competency interviews, deep-dive technical interviews covering programming and systems design, and a final mission and values alignment interview.

Are visa sponsorship and relocation support provided?

Yes, relocation support and visa sponsorship are provided where applicable.

Ready to apply?

The application is on the official wayve careers page.

Apply on company site