About the Role
Wayve is seeking a Senior Machine Learning Performance Engineer to join our Machine Learning Platform team. This role is pivotal in optimizing large-scale training jobs as we aim to scale our models to the next order of magnitude. The Machine Learning Platform team is responsible for our GPU training infrastructure and the software abstractions around it. Your primary focus will be on improving training efficiency.
Key Responsibilities
- Maximize MFU: Work on maximizing the Machine Fractional Utilization (MFU) of our large-scale training jobs.
- Code Profiling: Profile and identify bottlenecks in training code to enhance performance.
- GPU Kernel Implementation: Implement GPU kernels to improve training throughput.
- Collaboration: Work closely with research teams to integrate and test training efficiency improvements.
- Cluster Management: Own and improve our GPU training clusters.
About You
Essential Qualifications:
- Over 5 years of experience in performance optimization or machine learning engineering.
- Proven experience in optimizing large-scale training jobs on GPU compute clusters.
- Experience working in platform teams and collaborating with research teams.
- Ability to report and track benchmarked performance over time in an open and accessible way.
- Proficiency in writing high-quality, well-structured, and tested Python code.
- BS or MS in Machine Learning, Computer Science, Engineering, or a related technical discipline, or equivalent experience.
Desirable Skills:
- Solid experience with concurrent, parallel, and distributed computing.
- Experience using Nvidia NSight Systems.
- Experience implementing GPU kernels.
- Knowledge of computing fundamentals, including what makes code fast, secure, and reliable.
Why Join Us?
At Wayve, we are committed to creating a diverse, fair, and respectful culture that is inclusive of everyone based on their unique skills and perspectives. We value diversity, embrace new perspectives, and foster an inclusive work environment. Join us to tackle today's most complex challenges and pave the way for a smarter, safer future.
Work Environment
This is a full-time role based in our office in Mountain View, California. We operate a hybrid working policy that combines time together in our offices and workshops to fuel innovation, culture, relationships, and learning, with time spent working from home. We operate core working hours so you can determine the schedule that works best for you and your team.
If you’re passionate about self-driving cars and think you have what it takes to make a positive impact on the world, we encourage you to apply.
Benefits Extracted with AI
- Hybrid working policy
- Core working hours
Similar jobs
Last update: 23 minutes ago
Machine Learning Engineer - LLM Fine-tuning and Performance
Join NVIDIA as a Machine Learning Engineer specializing in LLM fine-tuning and performance optimization. Work with cutting-edge ML technologies.
Expert Machine Learning Engineer
Join Dataroots as an Expert Machine Learning Engineer to design and deliver AI-powered solutions, focusing on machine learning models.
Software Engineer - Autonomous Driving
Join Applied Intuition as a Software Engineer in Munich to tackle autonomous driving challenges with top ADAS/AV programs.
Senior Backend Engineer C++
Join DeepL as a Senior Backend Engineer C++ to design and maintain scalable backend services using C++ and AI technologies.
Senior Backend Engineer - Java, Rust, Go
Join Together AI as a Senior Backend Engineer in Amsterdam. Work with Java, Rust, and Go to build scalable backend systems.
Principal AI Engineer
Join Cere Network as a Principal AI Engineer to drive AI innovation in Web3. Requires 10+ years in AI/ML, NLP, and software development.
Senior Backend Engineer (Go)
Join Nebius as a Senior Backend Engineer (Go) to develop fault-tolerant cloud services in a hybrid work environment.
Information Retrieval Algorithm Engineer
Join Huawei as an Information Retrieval Algorithm Engineer to develop cutting-edge AI technologies in Amsterdam.
Senior Software Engineer - AWS, Python, Ruby on Rails
Join HeyJobs as a Senior Software Engineer to design scalable systems using AWS, Python, and Ruby on Rails in a dynamic team.
Staff Software Engineer
Join Aiven as a Staff Software Engineer to develop cloud operations platforms using open-source technologies. Hybrid work in Berlin.
Senior AI Engineer
Join Poggio as a Senior AI Engineer to innovate AI systems for enterprise sales, focusing on AI capabilities and system performance.
AI Engineer
Join BCG X as an AI Engineer in Milan, Italy. Develop AI solutions, partner with clients, and drive innovation in a dynamic environment.
Senior Software Engineer (C++)
Join Nebius as a Senior Software Engineer (C++) to develop reliable cloud services in a hybrid work environment.
Cloud Data Engineer
Seeking a Cloud Data Engineer with expertise in AWS, Python, and CI/CD for a hybrid role in Hannover. Join our dynamic team!
Machine Learning Engineer
Join Pipedrive as a Machine Learning Engineer in Tallinn to deploy and optimize ML models, ensuring performance and compliance.
Back End Engineer with Node.js and TypeScript
Join LegalFly as a Back End Engineer to revolutionize legal AI with Node.js and TypeScript in a hybrid role in Ghent.
Senior Backend Developer
Join Motius as a Senior Backend Developer to work on cutting-edge R&D projects using AWS, Docker, GraphQL, and more in a hybrid work environment.
Senior Software Engineer - LLM
Join Bonfy.AI as a Senior Software Engineer to develop and optimize scalable machine learning models using Python, TensorFlow, and cloud platforms.
Machine Learning Engineer
Join MoonPay as a Machine Learning Engineer to build and maintain ML infrastructure, collaborating with data scientists and cross-functional teams.
Senior Software Engineer (C++, Python & Cloud)
Join VIAVI Solutions as a Senior Software Engineer specializing in C++, Python, and cloud technologies. Work in a hybrid environment in Berlin.
Senior Software Engineer - Java, Microservices
Join Sofico as a Senior Software Engineer focusing on Java and Microservices in Bavaria, Germany. Work on ERP solutions for automotive finance.
Senior Software Engineer - Java, Microservices
Join Sofico as a Senior Software Engineer focusing on Java and Microservices in Ghent. Work on ERP solutions for automotive finance and leasing.
AI Engineer - Machine Learning and Robotics
Join Blueprint as an AI Engineer in Machine Learning and Robotics, focusing on scalable AI model training systems. Hybrid role in Redmond, WA.
Senior Python Developer with AWS Experience
Join Basetime BV as a Senior Python Developer to develop and maintain AWS cloud solutions. Hybrid work, competitive salary, and growth opportunities.
