Staff Machine Learning Engineer, Vision Models

Stack

Responsibilities

• ▹Build, train, and fine-tune scene understanding models at the centre of Wayve's offline measurement, adapting on-vehicle architectures and Wayve Foundation Models for offline use
• ▹Drive accuracy and generalisation across vehicle platforms, geographies, and driving conditions; diagnose failure modes
• ▹Exploit the offline environment: higher compute budgets, larger model capacity, bidirectional temporal context, multi-task learning
• ▹Benchmark your own models, set quality bars, and use metrics and error analysis to steer the next iteration
• ▹Ensure benchmarked results are statistically defensible and fit to feed validation pipelines and safety cases
• ▹Align priorities and mentor across on-vehicle modelling, evaluation, data curation, and simulation teams

Requirements

• ▹5+ years in ML engineering, including training and shipping deep learning models in production
• ▹Hands-on experience training modern computer vision models, including transformer-based and multimodal/VLM architectures for detection, segmentation, classification, or scene understanding, on camera and/or lidar data
• ▹Experience adapting or fine-tuning large pretrained/foundation models, training shared representations across multiple tasks
• ▹Proficient in Python and ML frameworks (esp. PyTorch), with solid software engineering practices and large-scale training experience
• ▹Staff-level technical leadership: research-literate and pragmatic, setting direction without formal line management
• ▹Able to measure your own models: defining and reading metrics that show genuine improvement

Nice to have

• ▹Experience in 3D scene understanding and representation learning for geometric and semantic perception
• ▹Experience with offboard/offline modelling: auto-labelling, model distillation, temporal or world models
• ▹Prior experience in autonomous vehicles or robotics with hands-on deployment and closed-loop validation
• ▹Experience with fleet-scale data and large-scale distributed training infrastructure

Soft skills

What we offer

• ▹Competitive equity package
• ▹Hybrid working policy based out of the Sunnyvale office

About the company

Wayve's Measurement team, part of AI Evaluation, builds and qualifies the scene understanding models used to measure the Wayve Driver's performance offline.