Machine Learning Performance Engineer

Wayve, founded in 2017, is at the forefront of Embodied AI technology for autonomous vehicles. We're seeking a Machine Learning Performance Engineer to join our Machine Learning Platform team, focusing on optimizing large-scale training jobs as we scale our models.

Key responsibilities include:

• Maximizing the MFU of large-scale training jobs
• Profiling and identifying bottlenecks in training code
• Implementing GPU kernels to improve training throughput
• Collaborating with Research teams to integrate and test efficiency improvements
• Managing and enhancing our GPU training clusters

The ideal candidate will have:

• 5+ years of experience in performance optimization or ML engineering
• Expertise in optimizing large-scale training jobs on GPU compute clusters
• Experience working in platform teams and with research teams
• Proficiency in benchmarking and reporting performance metrics
• Strong Python coding skills
• BS or MS in Machine Learning, Computer Science, Engineering, or related field

Desirable skills include experience with concurrent and distributed computing, Nvidia NSight Systems, GPU kernel implementation, and a deep understanding of computing fundamentals.

At Wayve, we value diversity and inclusivity. We offer a hybrid working model, combining office time for innovation and collaboration with the flexibility of working from home. Join us in creating autonomy that propels the world forward!