Staff ML Engineer, Gaia
Full-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 Gaia is Wayve’s video world model: trained on large-scale driving video, it predicts future frames from past context—functioning as a simulator that helps generate synthetic scenarios, including rare or safety-critical events. As a Staff ML Engineer on Gaia, you’ll own and drive work on training and improving frontier-scale models trained in-house. This is a high-impact role with the opportunity to tech-lead a key area and help shape the next version of Gaia in a fast-paced, results-focused environment. Key Responsibilities
- Lead and execute large-scale training runs for video (or adjacent) foundation models, from experimental design through production-grade execution
- Contribute to model architecture and training strategy, using first-principles understanding rather than “off-the-shelf” application
- Improve world-model capabilities that enable synthetic scenario generation and downstream evaluation/training of the driving model
- Partner closely with research, applications, simulation engineering, and cloud/infrastructure teams to deliver end-to-end impact
- Provide technical leadership through mentorship, review, and setting high engineering/research standards (Senior/Staff scope)
- In-depth experience training large-scale models (language, video, or other foundation models), including ownership of training at scale
- Strong understanding of model architecture and the ability to contribute meaningfully to architectural/training decisions
- Strong hands-on engineering skills with modern ML stacks (e.g., PyTorch), including debugging and performance/reliability-minded development
- Relevant industry experience (typically 4–5+ years); advanced degrees are valued, but depth of applied experience is important
- Direct experience with world models, video generation, or long-horizon prediction
- Experience improving data/training pipelines and working across infrastructure constraints (distributed training, efficiency, reliability)
- Proven technical leadership (tech lead ownership, mentoring, setting direction across an area)
Vacancy posted 1 day ago
Similar jobs that could be interesting for youBased on the Staff ML Engineer, Gaia in United Kingdom vacancy
- ...continuously improve the platform and our day to day workflows, rather than treating these as secondary projects. As a Staff Machine Learning Engineer, you will hold a hands-on technical position. You will be part of a team that builds, ships, and maintains the entire...SuggestedFull timeWork at officeRemote workRelocationHome officeFlexible hours2 days per week1 day per week
- ...scale, and optimise next-generation world model architectures (GAIA and successors) and bridge them into high-throughput... ...same training stack we use for real driving. You sit between ML research and engineering: collaborating with scientists on architecture and conditioning...SuggestedFull timeWork at officeWork from home
- ...remote, async. Nice to have Solana validator operations / staking infrastructure. MEV ecosystem — relayers, bundles, block engines. SUI, Hyperliquid, or other high-throughput chains. Real-time market-data / streaming systems at scale. Low-level...SuggestedFull timeFor contractorsRemote workOverseas
- ...and one where everyone is valued. We strive for excellence in everything we do. The Role XTX is seeking an experienced engineer to support our ML Performance and AI Acceleration function. You will be working in a small and highly technical team at extremely high- and...SuggestedFull timeWork at officeWorldwide
- ...outdoors and a desire to protect it for future generations. Role Summary We are looking for a full-time Machine Learning Engineer, with deep knowledge and strong enthusiasm towards establishing a state-of-art AI infrastructure for training very large...SuggestedFull timeContract workLocal area
- ...businesses think about cybersecurity, digital experiences, and identity and access management. As a Ping Identity Staff Site Reliability Engineer, you will be involved in every facet of our Cloud-based services. You will establish solutions for building, deploying...Full timeLocal areaWorldwideFlexible hours
- ...prediction models across ARQ’s countries and acquisition channels. Work closely with Growth & Marketing, Product, Finance, Data Engineering, and Engineering teams to understand business needs and translate them into modelling solutions. Help evaluate acquisition...Full timeWork at office3 days per week
- ...take problems from idea to production — partnering with Product, Engineering, and Operations teams to turn large-scale datasets into models... .... Transform ambiguous business problems into measurable ML solutions. Partner closely with Product and Operations teams...Full timeWork at office3 days per week
- ...do is the result of the creativity and commitment of our team - builders doing the best work of their lives. We are researchers, engineers, and operators. IOI medalists and ex-founders. If you want to work hard and create lasting positive impact, we want to hear from you...Full timeImmediate startRemote work
- ...understand the underlying behaviours employed by criminals. You will be product focused, working in close collaboration with our engineering and operations data scientists as well as the wider sales, consulting, and product teams. In this position, you will: Perform...Full timeWorldwide
- ...to the team… Are pursuing a bachelor’s, master’s or PhD in mathematics, economics, physics, statistics, computer science or any engineering related field and have an expected graduation date between December 2027 and August 2028 Advanced quantitative, analytical...Full timeWork experience placementSummer workInternshipWork at officeImmediate startDay shiftAfternoon shift
- ...partnership with Ingenico bringing stablecoin payments to 40M+ terminals across 120+ countries. The Role This is a Senior Data Engineer position on the Data Engineering team. You’ll work closely with the Data Engineering Lead to design and scale the data platform...Full timeWork at officeRemote workWork from homeHome office
- ...We're hiring a Staff QA Automation Engineer to help shape how quality engineering works across tem. QA sits inside Platform, a central team that builds the tooling and standards the rest of engineering runs on. You'll design the frameworks and practices that let them...Full timeContract work
- ...integrating a high-performance central limit order book matching engine with automated market making to provide deep and predictable... ...+ years in backend development, with a proven track record in a Staff, Principal, or equivalent technical leadership role where you were...Full time
- ...-million ARR company with a team of 14 from AWS, Snyk, GitHub, Okta, Netlify, Uber, Samsara, and more. We are hiring a Staff Platform Engineer to own the infrastructure Nango runs on - and increasingly, the infrastructure our customers run Nango on. This is a critical...Full timeRemote work
$80 - $150 per hour
Physics Expert (Postdoc / Junior professor) Pay: $80 - $150/hour Role Type: Contractor Location: Remote (US, Canada, UK focused) This remote contractor role engages Physics Experts at the postdoc or junior professor level for a high-impact project supporting...Remote jobFor contractors- ...partnering across Credit Risk, Product, Engineering, Analytics, and Commercial teams to shape... ...elegant models. Comfortable operating at staff level: you can set direction in ambiguity... ...teams to deploy decisioning or ML systems into live products, rather than operating...Full timeWork at officeImmediate start
- ...experienced, technical, and hands-on Autonomy Data Manager to grow and lead a talented team composed of Data Scientists, Data Engineers and ML Ops Engineers, working in the Autonomy team to deliver data driven Planning and Perception solutions. You will lead the team to...Full timeTemporary workWork at officeFlexible hours
- ...reliable production workflows. ~ Python: You are comfortable using Python to build pipelines, automate processes, and develop data engineering tools. ~ Data modeling: You understand dimensional modeling, ELT patterns, and how to design datasets for reliability,...Full timeRemote work
- ...We are seeking a Senior Analytics Engineer to drive significant business impact by shaping our data infrastructure, developing robust data models, and promoting best practices in analytics engineering (e.g., data quality, documentation, and scalable data architectures...Full time
- ...enables consistent KPI definitions and supports downstream consumers, including LLM-based analytics agents. Collaborate with Data Engineers on upstream data contracts and event schemas - ensuring raw data is structured in a way that supports scalable, reliable...Full time
- ...Your Next Destination We're looking for a Staff Data Scientist, Pricing to build the... ...across the full lifecycle, from feature engineering and model selection through to deployment... ...the decision. ~ Fluency in the Python ML stack alongside strong SQL , with the...Full time
$1,000 per month
...Do you want to help define the future of finance? Are you a strong software engineer who's passionate about data, distributed systems, and delivering impact at scale? Are you looking for a values-driven company that invests in its people and gives you the autonomy to...Remote jobFull timeHome office- From empowering entrepreneurs to build successful businesses, to supporting their clients to arrange 'me time' moments, we’re in the business of elevating people - helping them to thrive and feel fantastic. Lifting others up can be hard, heavy work, and we passionately ...Full time
- ...to grow your career at Booksy. The people you’ll like to work with and things you'll enjoy impacting: As an Associate Data Engineering Manager reporting to the Head of Data Engineering in our Data & Analytics - Data Engineering team, your purpose will be to lead a...Full timeWork at officeRemote work
- ...via Claude, return accurate, contextually aware results. You are the steward of how well our BI tool thinks. Partner with data engineering to optimise performance, and build what's missing Own and maintain the semantic data layer and transformation logic so that...Full time
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Staff ML Engineer, Gaia. Be the first to apply!
Related searches
- data engineer machine learning United Kingdom
- machine learning United Kingdom
- information technology administrative assistant
- staff engineer
- assistant engineer
- assistant shift engineer
- assistant resident engineer
- engineering administrative assistant
- graduate assistant engineering
- research assistant engineering










