Staff Machine Learning Engineer, Vision Models
$370.04k - $407.33kFull-time
Wayve
About Us Founded in 2017, Wayve is the leading developer of Embodied AI technology. Our advanced AI software and foundation models enable vehicles to perceive, understand, and navigate any complex environment, enhancing the usability and safety of automated driving systems. Our vision is to create autonomy that propels the world forward. Our intelligent, mapless, and hardware-agnostic AI products are designed for automakers, accelerating the transition from assisted to automated driving. In our fast-paced environment big problems ignite us—we embrace uncertainty, leaning into complex challenges to unlock groundbreaking solutions. We aim high and stay humble in our pursuit of excellence, constantly learning and evolving as we pave the way for a smarter, safer future. At Wayve, your contributions matter. We value diversity, embrace new perspectives, and foster an inclusive work environment; we back each other to deliver impact. Make Wayve the experience that defines your career! The role As a Staff Machine Learning Engineer on Wayve's Measurement team in AI Evaluation, based in our Sunnyvale office, you will build the computer vision and scene understanding models Wayve uses to measure the performance of the Wayve Driver offline. You will adapt technology from our on-vehicle models and Wayve Foundation Models into offline models that understand coverage, mine rare events, and assess driving behaviour, and you will drive their accuracy and generalisation across vehicles, markets, and conditions. Measuring your own models rigorously is part of the work. You will define ground truth and correctness criteria across a complex driving taxonomy, and turn them into automated benchmarks and evidence that our validation pipelines and safety cases can stand on. The Measurement team builds and qualifies the scene understanding models Wayve uses to measure driving performance offline, after on-road runs and in simulation. Offline is where the interesting headroom is: more compute per frame, larger foundation models, and access to both past and future temporal context that the vehicle never has. The outputs are mission-critical, directly informing model development decisions and customer deliverables. You will work in a focused, high-impact senior team with strong ownership, access to fleet-scale camera, lidar, and simulation data, and close partners across on-vehicle modelling, evaluation, data curation, and simulation. Key Responsibilities
- Develop the models - build, train, and fine-tune the 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 - improve model performance across vehicle platforms, geographies, and driving conditions; diagnose failure modes and close the loop on blind spots.
- Exploit the offline environment - use the advantages the vehicle does not have: higher compute budgets, larger model capacity, bidirectional temporal context, and multi-task or joint representation learning.
- Measure what you build - benchmark your models, set quality bars, and use metrics and error analysis to steer the next iteration; treat measurement as the feedback that drives the modelling.
- Make the evidence credible - ensure benchmarked results are statistically defensible and fit to feed validation pipelines at scale and our broader safety cases, across the product portfolio.
- Align priorities and mentor - work day-to-day with on-vehicle modelling, evaluation, data curation, and simulation teams across sites; raise the bar on engineering and modelling practice; mentor others on the team; keep sight of division and company priorities and how Measurement work enables them.
- 5+ years in ML engineering, including training and shipping deep learning models in production, with pathfinding in ambiguous modelling problems from scoping through to a direction others build on.
- Hands-on experience training modern computer vision models, including transformer-based and multimodal or VLM architectures for detection, segmentation, classification, or scene understanding, on camera and/or lidar sensor data.
- Experience adapting or fine-tuning large pretrained or foundation models, and training shared representations across multiple tasks or objectives (multi-stage or joint training), including real trade-offs across data and losses.
- Proficient in Python and ML frameworks (esp. PyTorch), with a solid foundation in software engineering practices and comfort with large-scale training.
- Staff-level technical leadership: research-literate and pragmatic, setting direction, raising the bar, and leading cross-functional work without formal line management.
- Able to measure your own models: comfortable defining and reading the metrics that show whether a model is genuinely improving.
- Experience in 3D scene understanding and representation learning for geometric and semantic perception, including large-scale semantic enrichment of driving scenes.
- Experience with offboard or offline modelling: auto-labelling, model distillation, temporal or world models, or other ways of exploiting compute that on-vehicle systems cannot.
- Prior experience in autonomous vehicles or robotics with hands-on deployment and closed-loop validation on physical systems.
- Experience with fleet-scale data and large-scale distributed training infrastructure.
Vacancy posted 24 days ago
Similar jobs that could be interesting for youBased on the Staff Machine Learning Engineer, Vision Models in California vacancy
$251k - $310k
...the ultimate cognitive engine for autonomous driving.... ...multimodal foundation models (e.g., Gemini) to build... ...will report to a Senior Staff Technical Lead Manager.... ...SFT) and Reinforcement Learning (RLHF/RLAIF, DPO/GRPO/PPO... ...Models (LLMs) or Vision-Language Models (VLMs)...SuggestedFull timeRemote work$161.26k - $332.01k
Pinterest is seeking a skilled Research Engineer to join our visual modeling team focusing on generative models, including text-to-image development.... ...Candidates should have significant experience in computer vision and strong background in diffusion models. The role...Suggested$250k - $295k
...use advanced physics and AI to model catastrophic risk at the... ...real product is a scalable risk engine, our Stand World Model . We... ...The Opportunity: As a Machine Learning Engineer on the Applied Science... ...heterogeneous data (e.g., 3D/vision, simulation outputs, tabular,...SuggestedFull timeTemporary workH1bWork at officeRemote workVisa sponsorshipWork visaFlexible hoursStand Insurance
San Francisco, CA1 day ago$160k - $300k
...and introducing a Factory-as-a-Service model that delivers complete systems instead... ...for the factory.As a Senior Machine Learning Engineer, you will own the ML lifecycle for the... ...experience building and shipping computer vision models, with an emphasis on detection...SuggestedPermanent employmentFull timeLocal areaFlexible hours- ...—including core computer vision, multimodal representation learning, heterogeneous graph neural networks, generative modeling, and recommender systems.... ...pod is a small group (~10 engineers) inside Labs, which... ...team. M.S. or PhD in Machine Learning or related academic...SuggestedFull timeWork experience placementCurrently hiringWork at officeRemote workRelocationRelocation packageFlexible hours
Pinterest
San Francisco, CA1 day ago - .... Develop and integrate large language models (LLMs) and other state-of-the-art AI techniques... ...experiences. Knowledge and passion in machine learning algorithms, Gen AI, LLMs, and natural... ...benefits with family medical, vision and dental coverage, a competitive base...Full timeWork experience placement
Eightfold
Santa Clara, CA1 day ago $170k - $216k
...builds the system which learns the spatial-temporal representation... ...set of sensors, enabling engineers like you to (1) develop... ...data, to (2) develop models and model training at... ...~3+ years experience in Machine Learning and/or Computer Vision ~ Experience with Python...Full timeRemote workWaymo
San Francisco, CA1 day ago$203k - $270k
...former U.S. Navy electrical engineers with deep experience in... ...together AI, computer vision, precision motion, and... ...We are looking for a Staff Level Computer Vision and Machine Learning Engineer to help lead the... ...implement machine learning models that can operate in...Local area$229k - $343k
...themselves, live in the moment, learn about the world, and have... ...AI, including foundational models, efficient infrastructure, and... ....We’re looking for a Machine Learning Engineer to join Snap Inc!What you’ll... ...experienceExperience with Computer Vision or Generative Modeling...Full timeLive inWork at officeLocal areaWorldwide$244.14k - $413.16k
...through cutting-edge R&D in AI, machine learning, and smart connectivity.The... ...we are developing large-scale Vision-Language-Action (VLA) models and World Models to handle the... ...of global driving. As a Senior Staff Machine Learning Engineer, you will architect the transition...Full timeOverseas$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... ...team of experts in computer vision, generative AI, and AI systems, powered...Full time$174.72k - $295.68k
...transportation 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 XPENG’s next-generation Vision-Language-Action (VLA) Foundation...Full time$232k - $310k
...collaboration, and high standards. Our engineers, product leaders, and go-to-... ...the boundaries of applied machine learning. We work with massive... ...cutting-edge deep learning models across all Eightfold... ...benefits with family medical, vision and dental coverage, a competitive...Work experience placementWork at officeRemote workFlexible hours3 days per week$298k - $368k
...builds the system which learns the spatial-temporal... ...set of sensors, enabling engineers like you to (1) develop... ...world data, to (2) develop models and model training at... ...years of experience in Machine Learning, with a focus... ...learning or computer vision conferences (e.g., NeurIPS...Full timeRemote workWaymo
San Francisco, CA1 day ago$213k - $263k
...stack. We are an advanced ML and engineering team that leverages state-of-the-art computer vision, deep learning, and generative AI to... ...you will report to a Senior Staff Technical Lead Manager. You... ...computer vision / multimodal models (e.g., Gemini) to extract the...Full timeRemote workWaymo
San Francisco, CA3 days ago$227.2k - $284k
...that power the world’s most advanced models — fueling breakthroughs in... ...agentic AI looks like.About the RoleAs a Staff Machine Learning Research Engineer, you will operate across the full breadth... ...: comprehensive health, dental and vision coverage, retirement benefits, a...Full time$190k - $253k
Matterport – Senior Machine Learning/Computer Vision Engineer – 3D Reconstruction and Semantic Understanding Job Description Company Overview:CoStar... ...and output quality.Apply deep learning and foundation model techniques across the vision pipeline — including...Full timeRemote work- ...Institute of Foundation Models We are a dedicated... ...data scientists, and engineers, tackling the most fundamental... ...computing in deep learning, driving impactful... ...foundation models to unlock machine intelligence beyond... ...who share our vision and are eager to push...Visa sponsorship
Institute of Foundation Models
Sunnyvale, CA12 days ago - ...philosophy and how we use AI in our recruiting process here . We are looking for a Staff Machine Learning Engineer to lead the technical vision for our Ads Conversion Core Modeling team, building the state-of-the-art systems that power our global marketplace....Full timeWork at officeRelocationRelocation package
Pinterest
San Francisco, CA1 day ago $229k - $343k
...themselves, live in the moment, learn about the world, and have... ...AI, including foundational models, efficient infrastructure, and... ....We’re looking for a Machine Learning Engineer to join Snap Inc!What you’ll... ...experienceExperience with Computer Vision or Generative Modeling...Live inWork at officeLocal areaWorldwide$407.33k
The role As a Principal Engineer on the Model Foundations team you will build the geometric vision and 3D foundation models that underpin... ...of large-scale deep learning, geometric computer vision, and... ...up for success as a Principal Machine Learning Engineer, Geometric...Full timeWork at officeWork from home- ...world’s first post-transformer model that adapts and thinks just... ...the fastest data processing engine on the market, Pathway enables... ...with a strong track record in machine learning models research . This is an... ...and/or RL (candidates with a Vision or Robotics ML background are...Permanent employmentFull timeContract workImmediate startRemote workFlexible hours
$230k - $322k
...Team overview: The Engagement Modeling Team at Reddit focuses on building machine learning models to drive on-platform user... ...learning expertise, strategic vision, and a collaborative mindset to... ...PMs, data scientists, and other engineering teams to align on engagement strategies...For contractorsWork experience placementWork at officeRemote workHome officeFlexible hoursGrabJobs
San Jose, 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... ...Science, Robotics, Machine Learning, or equivalent... ..., AI or Computer Vision. ~ Proven...Full timeRemote workShift workWaymo
San Francisco, CA1 day ago- ...AI software and foundation models enable vehicles to... ...automated driving systems. Our vision is to create autonomy that... ...of excellence, constantly learning and evolving as we pave the... ...your career! The role As a Staff Machine Learning Engineer in Wayve's AV Core...Shift work
$160.5k - $240.7k
...Technologies, Inc. Job Area Engineering Group, Engineering Group > Machine Learning Engineering About the... ...deploy machine learning models on edge and mobile... ...compressing latency-critical vision, audio, and multimodal networks... ...talented and motivated Staff Software Engineer with...Work experience placementImmediate startWork from home$168.1k - $312.3k
...leveraging data and novel computational models to drive impact. Seamless data sharing and... ...Our mission is to develop AI models that learn from biological data at unprecedented scale... ...motivated and passionate Senior ML Engineer to join our Generative Modeling team and...Full timeLocal areaWorldwideRelocation package$94k - $270k
...U.S. Navy electrical engineers with deep experience in... ...together AI, computer vision, precision motion, and... ...Computer Vision and Machine Learning Engineer to develop and... ...Associate through Senior Staff; your title and level... ...machine learning models capable of operating in...Local areaAllen Control Systems
Mountain View, CA12 days ago- A leading spatial computing firm is seeking a Staff Machine Learning R&D Engineer to innovate their platform. This role involves developing machine learning algorithms for 3D data processing and computer vision. Candidates should have a strong background in ML research...
- ...fully autonomous Corvus One™ drones use computer vision & robotics to automatically track inventory,... ...About the role We are hiring Computer Vision / Machine Learning Software Engineers to build compute-constrained models for deployed robots. You'll tackle diverse technical...Visa sponsorship
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Staff Machine Learning Engineer, Vision Models. Be the first to apply!
Related searches
- engineering aide California
- senior staff engineer California
- assistant engineer California
- software engineer staff California
- staff engineer California
- senior staff systems engineer California
- technology administrator California
- machine learning engineer California
- computer vision machine learning engineer California
- senior ml engineer California