Lead end-to-end development of production machine learning systems for cash-flow underwriting, shaping technical direction and turning transaction data into fair, accurate credit decisions and measurable member outcomes.
Staff Engineer, World Model Development
📜 Description
- Design, train, and evaluate world models predicting aircraft state, environment, and traffic under candidate actions.
- Build rollout tools that let planners evaluate candidate plans against predicted outcomes.
- Provide calibrated uncertainty and out-of-distribution signals for safety monitoring.
- Define where predictions are reliable across flight regimes, weather, traffic densities, and configurations.
- Train models in simulation, validate against flight data, and work with the Data/Sim team to improve simulator fidelity.
- Benchmark learned dynamics against physics-based and flight-controls models, and maintain reproducible evaluations.
🛠️ Requirements
- Degree in Computer Science, Artificial Intelligence, Data Science, Computer Engineering, Applied Math, or a related subject.
- 10+ years building and training deep learning models, with meaningful work on sequence, dynamics, video or trajectory prediction.
- Strong PyTorch and a solid grasp of the training stack — distributed training, mixed precision, experiment tracking, debugging a run that has silently gone wrong.
- Working knowledge of dynamical systems, state estimation or control; you can read a flight dynamics model and know what your network is and is not replacing.
- Demonstrated rigor in evaluation and uncertainty quantification.
- Comfort with messy real-world sensor and telemetry data.
- Model-based RL, latent dynamics models or learned simulators.
- Experience with trajectory prediction in autonomous driving or robotics.
- Aerospace background: flight dynamics, aircraft performance or air traffic.
- Multimodal fusion across vision, state and structured mission context.
✨ Benefits
- Health benefits
- Dental benefits
- Life insurance
- Flexible paid time off
- 401(k) with match
- Work/life integration
Full job description
About You:
Responsibilities:
Qualifications:
Nice to Have:
Logistics:
Similar jobs
Search more Machine Learning Engineer jobsLead post-training and closed-loop development for Gaia, Wayve’s driving world model, improving long-horizon generation, model reliability, and downstream simulation impact through experimentation, data curation, and technical direction.
Machine Learning Engineer, Application Software
Lead the development and production deployment of machine-learning driving behaviours, improving autonomous driving performance through model architecture, data pipelines, evaluation, and cross-team technical leadership.
Staff Machine Learning Engineer - VoIP Infrastructure
Designs and operates VoIP infrastructure for AI-driven voice workloads, integrating telephony platforms with LLM features while improving reliability, observability, deployment practices, and production support.
Staff Machine Learning Engineer - VoIP Infrastructure
Design and operate VoIP infrastructure for AI-driven voice workloads, integrating telecom platforms with LLM features while improving reliability, observability, and production support across distributed systems.
