Principal Machine Learning Engineer GAIA

Added
4 hours ago
Type
Full time
Salary
Salary not provided

Related skills

pytorch machine learning diffusion models 3d vision world models

πŸ“‹ Description

  • Own and lead parts of large-scale training for Gaia (language- or video-style training at scale)
  • Contribute to and influence model architecture decisions β€” not just applying existing models.
  • Partner across sub-teams: Research (model development), Applications (adapting models to use
  • Translate product and evaluation needs into model improvements (e.g. better long-horizon
  • Mentor and set technical direction, raising the engineering and research quality bar across the
  • Lead and execute Gaia's post-training and closed-loop pipeline β€” fine-tuning and aligning the world

🎯 Requirements

  • Hands-on experience post-training/fine-tuning large-scale models (language, video, or other
  • Experience with world models, autoregressive generation, and long-horizon generation.
  • Experience with diffusion/flow models and a solid understanding of 3D vision.
  • Strong understanding of model architecture and can contribute meaningfully to
  • Strong hands-on engineering skills with modern ML stacks (e.g. PyTorch), including debugging and
  • 5+ years of relevant industry experience (advanced degrees are valued, but depth of applied

🎁 Benefits

  • Salaries benchmarked against the market annually
  • Meaningful equity, sharing in the ownership and long term success of Wayve
  • Relocation support and visa sponsorship where applicable
  • Hybrid working, core hours and the chance to work hands on in vehicle workshops and labs
  • Learning and development budgets with support for training, conferences and growth
  • Comprehensive benefits including health insurance, dental, enhanced maternity and paternity leave

🚚 Relocation support

πŸ›ƒ Visa sponsorship

Get alerts for new remote engineering jobs
A weekly email with new matches. Unsubscribe anytime.
Share job

Meet JobCopilot: Your Personal AI Job Hunter

Automatically Apply to Engineering Jobs. Just set your preferences and Job Copilot will do the rest β€” finding, filtering, and applying while you focus on what matters.

Related Engineering Jobs

See more Engineering jobs β†’