Senior Machine Learning Engineer, Vision Models
$312k - $389k • Sunnyvale, California USA • Full-time
Posted 28d ago
Remote Work Policy
On-site
Employment Type
Full-time
Categories
Machine Learning Engineer
About the job
Wayve is building the AI platform for autonomous driving, enabling vehicles to learn directly from real-world experience. This role is on the Measurement team within AI Evaluation, focused on building and qualifying offline scene-understanding models that measure driving performance. These models leverage greater compute, larger models, and temporal context to create credible evidence at fleet scale, contributing to the development of a generalizable AI Driver for real vehicles.
Responsibilities
- Build, train, and fine-tune scene-understanding models for offline measurement.
- Adapt on-vehicle architectures and foundation models for offline use.
- Improve model accuracy and generalization across different vehicle platforms, geographies, and driving conditions.
- Benchmark models, diagnose failure modes, and use error analysis for iteration.
- Define quality bars and statistically credible evidence for validation pipelines and safety cases.
- Collaborate with on-vehicle modeling, evaluation, data curation, and simulation teams.
- Mentor other team members and contribute to strong machine learning and engineering practices.
Requirements
- 4+ years in ML engineering with experience training and shipping deep learning models in production.
- Hands-on experience with modern computer vision, transformer, multimodal, or VLM architectures.
- Experience adapting or fine-tuning large pretrained or foundation models, understanding data, loss, and representation trade-offs.
- Proficiency in Python and PyTorch or a similar framework, with comfort in large-scale training.
- Ability to take ambiguous modeling problems from scoping to a working, measurable solution.
- Experience in 3D scene understanding, offline modeling, auto-labeling, temporal or world models, autonomous vehicles, robotics, or distributed training is valuable.
Benefits
- Salaries benchmarked against the market annually
- Meaningful equity
- Relocation support
- Visa sponsorship
- Hybrid working
- Access to vehicle workshops and labs
- Learning and development support
- Comprehensive health, family, retirement, and wellbeing benefits