Principal Machine Learning Engineer, 3D Geometric Vision

Job description

Before the detail, here's the challenge you'd help us solve. We build the embodied intelligence that moves real vehicles safely, and the ecosystem a billion machines will run on in the future. Very few people in AI can say this. Every role here, whatever the team, plugs into that. Here’s what this particular role covers. 🛠️ About our Teams The Model Foundations team builds the geometric vision and 3D foundation models that underpin Wayve's autonomous driving systems, working at the intersection of large-scale deep learning, geometric computer vision, and real-world robotics to develop models that learn 3D structure and dynamics from fleet-scale sensor data. 🧠 Your day-to-day As a Principal Engineer on the Model Foundations team, you will be a hands-on technical leader. You'll set direction for geometric vision, prototype and train new model architectures, build the data and supervision needed to scale them, and take successful ideas through to deployment on real vehicles. 🧩 What you’ll be working on - 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, using large-scale sensor data to recover accurate trajectories, scene geometry, calibration signals, and geometric supervision for model training and evaluation. - Develop and apply techniques in multi-view geometry, neural rendering, NeRFs, Gaussian Splatting, implicit 3D representations, and feedforward 3D modeling. - Explore geometry-aware tokenization and representation learning, including efficient ways to encode and fuse information across cameras, viewpoints, time, and sensing modalities. - Develop foundation vision models that make effective use of camera, radar, LiDAR, and other sensor data for learning rich representations of the physical world. - Explore video and generative modeling approaches for learning scene structure, dynamics, and future evolution from driving data. - Train and evaluate models at scale on distributed compute, rapidly iterating on architectures, objectives, data, and training recipes. - Develop automated evaluation and ground-truth systems for measuring geometric consistency, reconstruction quality, 3D understanding, and downstream driving performance. - Optimize and deploy models into production autonomous-driving systems, working across model architecture, inference, and onboard constraints. - Set technical direction for geometric vision at Wayve and work closely with researchers and engineers across foundation models, perception, simulation, data, sensing, and deployment. 🙌 You should apply if Essential - You have deep expertise in 3D computer vision, geometric vision, or 3D machine learning, with experience in areas such as multi-view geometry, neural rendering, reconstruction, implicit representations, or world modeling. - You have strong experience designing, training, and evaluating modern deep-learning models at scale, using PyTorch or a comparable framework. - You have strong mathematical and technical foundations in geometry, linear algebra, probability, optimization, and 3D transformations, combined with excellent software engineering skills in Python and C++. - You have a track record of taking difficult research problems from idea to working system, including building large-scale data, training, evaluation, or deployment pipelines. - You bring principal-level technical leadership: the ability to identify high-leverage problems, set research and engineering direction, make strong architectural decisions, and raise the technical bar across teams. Desirable - 3D and geometric vision: multi-view geometry, dense 3D reconstruction, neural fields, NeRFs, Gaussian Splatting, or feedforward 3D models. - Foundation and world models: large-scale vision pre-training, self-supervised learning, video models, generative models, or learned scene dynamics. - Geometric data engines: offline SLAM, structure-from-motion, reconstruction, calibration, auto-labeling, and large-scale ground-truth generation. - Multimodal perception: learned representations and fusion across camera, radar, LiDAR, and other sensing modalities. - Production ML systems: distributed training, large-scale experimentation, and deploying neural networks on real-time, resource-constrained hardware. 🌱 Not ticking every box? That’s totally okay! If you’re passionate about autonomy and keen to learn, we encourage you to apply even if you don’t meet every requirement. More about Wayve: 🚀 Wayve is building the leading AI platform for autonomous driving. We are pioneering an end to end AI approach that enables v