Wayve is seeking a Machine Learning Engineer to advance their Embodied AI technology, specifically focusing on enhancing the usability and safety of automated driving systems through state-of-the-art research in areas like Offline Reinforcement Learning, Reward Learning from Human Feedback, and large-scale embodied Foundation Models.
Requirements
- Solid experience with Python or proficiency in a systems/backend programming language with the ability to quickly adapt to Python.
- Demonstrated experience in system design, capable of architecting robust, scalable solutions.
- Experience with MLOps or ML Infrastructure.
- Experience with Offline Reinforcement Learning (RL).
- Experience with Reward Learning from Human Feedback (RLHF).
- Experience with large-scale, embodied Foundation Models.
- CUDA optimization
Responsibilities
- Collaborate with Applied Scientists and Machine Learning Engineers on advanced, multimodal, embodied Foundation Models.
- Develop and manage comprehensive datasets and data engineering pipelines.
- Craft and refine tools for rapid exploration and detailed visualizations.
- Drive observability, monitoring, and performance optimizations to elevate system reliability and performance.
- Create tools specifically designed to expedite solving research problems.
- Work seamlessly with platform teams, facilitating integration and leveraging shared resources for broader impact.
- Optimize pre-training runs, including data pre-processing, CUDA optimization, model quantization and optimization, and increasing throughput of training jobs (e.g., FP-8).
Other
- 7+ years of experience with a BS or MS in Computer Science, Engineering, or related discipline, or equivalent experience.
- Proven track record of working in teams to successfully deliver open-ended projects.
- Ability to work cross-functionally, bridging gaps between teams to drive collective goals.
- Champion engineering best practices, ensuring solutions are scalable, efficient, and maintainable.
- Prioritize code quality, readability, and reusability.
- Excel in ambiguous, fast-paced environments, adept at navigating and thriving amidst change.
- Passionate about self-driving cars and think you have what it takes to make a positive impact on the world.
