Principal Machine Learning Engineer GAIA
Core
Lead post-training and closed-loop pipelines for Gaia to improve long-horizon autoregressive generation, ensuring stable, coherent, and steerable rollouts.
Role type
Principal Machine Learning Engineer (Post-training & Model Architecture)
Builds
Post-training pipelines, fine-tuned models, and simulation engineering for long-horizon generation tasks.
Domain
Generative AI, World Models, 3D Vision, Large Language Models
Deliverable
production ML models
Required skills
Post-training and fine-tuning of large-scale foundation models, autoregressive generation, diffusion or flow models, 3D vision understanding, model architecture design, PyTorch, distributed training, inference optimization, large-model deployment under latency constraints, data curation, technical leadership
Preferred skills
Experience with world models, simulation engineering, cloud/infrastructure integration, efficiency optimization
Technologies
PyTorch
Responsibilities
Lead post-training and closed-loop pipeline including fine-tuning, alignment, and experimentation; Contribute to model architecture and training-strategy decisions; Improve inference optimization and deployment readiness; Provide technical leadership through mentorship and setting engineering standards
Seniority
Principal, hands-on IC with mentorship
