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 a month ago
Similar jobs that could be interesting for youBased on the Staff ML Engineer, Gaia in United Kingdom vacancy
- ...environment; we back each other to deliver impact. Make Wayve the experience that defines your career! The role As a Staff ML Performance Engineer, you’ll play a key role in high-impact projects, optimising ML inference for edge accelerators and GPUs. The focus of...SuggestedFull timeWork at officeWork from home
- ...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
- ...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...SuggestedRemote jobFull timeFor contractorsOverseas
- ...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
XTX Markets
United Kingdom2 days ago - ...Wayve the experience that defines your career! ** ️ About our Engineering Teams** Wayve’s ADAS engineering teams build the perception... ...based on clear evaluation signals. You’ll work across the full ML lifecycle (data → training → evaluation → iteration), partnering...SuggestedInternshipWork at officeRemote workVisa sponsorshipRelocation packageFlexible hoursShift work
- ...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...Full timeContract workLocal area
Rivian
United Kingdom14 days ago - ...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...Remote jobFull timeLocal areaWorldwideFlexible hours
$80 - $150 per hour
Physics Expert (Postdoc / Junior Professor) Remote contractor role focused on candidates in the United States, Canada, and the United Kingdom. Compensation is $80 to $150 per hour. Apply advanced physics expertise to a high-impact project supporting a customer in the...Remote jobHourly payFor contractorsSaidGig
United Kingdom3 days ago- ...defines your career! The Role As a Data Scientist supporting AI engineers, you will partner with one or more engineering teams, developing... ...in using a statistical scripting language and data science/ML packages (e.g. python such as pandas, sklearn, statsmodels, scipy...Full timeWork at officeWork from home2 days per week
- Our Purpose Mastercard powers economies and empowers people in 200+ countries and territories worldwide. Together with our customers, we’re helping build a sustainable economy where everyone can prosper. We support a wide range of digital payments choices, making transactions...Full timeWorldwide
- ...builders doing the best work of their lives. We are researchers, engineers, and operators. IOI medalists and ex-founders. If you want to... ...This is a hands-on role at the intersection of data science, ML, AI safety, and policy. You'll work closely with safety researchers...Remote jobFull timeImmediate start
- ...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...Remote jobFull timeWork at office3 days per week
- ...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...Remote jobFull timeWork at office3 days per week
- ...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...Remote jobFull timeWork at officeWork from homeHome office
£140 per hour
...Apply your production data engineering expertise to create real-world technical deliverables that help train and evaluate advanced AI models... ...in roles such as Data Engineer, Senior Data Engineer, Staff Data Engineer, Analytics Engineer, Data Platform Engineer, or Data...Hourly payRemote workSaidGig
United Kingdom8 days ago- ...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
- ...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...Remote jobFull timeWork at officeImmediate start
- ...value to our customers. We're looking for a Staff Data Scientist to join our Research and... ..., Global Intelligence, Product, and Engineering to move research from prototype to dependable... ...techniques (e.g. graph algorithms, ML, statistical inference) to large, messy datasets...Remote jobFull timeFlexible hours
- ...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
Valor Capital Group
United Kingdom2 days ago £76k - £114k per year
...usual support. Provide technical direction to Data Scientists, Engineers, MLOps specialists and Product Owners, ensuring solutions are... .... · ~ As a Santander UK employee, you are able to request staff versions of our products like our Edge Current Accounts and Credit...Full timePart timeFixed term contractWork experience placementWork at officeLocal areaWork from homeRelocationVisa sponsorshipFree visaFlexible hoursShift work
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Staff ML Engineer, Gaia. Be the first to apply!
Related searches
- machine learning United Kingdom
- machine learning remote United Kingdom
- staff process engineer
- structural engineering assistant
- software engineer staff
- information technology support assistant
- staff chemical engineer
- assistant field engineer
- assistant city engineer
- information technology administrative assistant