Machine Learning Engineer (Performance Tooling)

Join Wayve, a cutting-edge AI company, as a Machine Learning Engineer in the AI Performance Tooling team. In this role, you will build tools that optimize AI performance across various hardware and software platforms. You will work closely with multiple teams to turn performance measurements into actionable recommendations and set the standards for AI performance measurement at Wayve. Enjoy a competitive compensation package, private healthcare, paid time off, mental health resources, and opportunities for learning and development.

• Conception et développement d'outils de performance fiables et autonomes qui évoluent à travers les modèles, les cibles matérielles et les flux de travail de développement.
• Collaboration étroite avec les équipes de modèles, de compilateurs, d'exécution, de plateforme et de matériel pour transformer les mesures en recommandations claires.
• Responsabilité de la surveillance et de l'alerte de régression à travers les constructions de modèles et les exécutions d'entraînement.

• Hands-on experience developing deep learning models with PyTorch
• Judgment to turn an ambiguous performance question into a measurable one, and to prioritize what matters
• Deep, hands-on performance engineering in complex systems: profiling, roofline analysis, latency and throughput optimization, and root-causing what limits a workload
• Quantitative communication clear enough to influence another team’s priorities
• A track record of owning a tool or service end to end — design, delivery, and adoption by other teams
• Strong Python skills, and comfort profiling and instrumenting large production codebases
• Data analysis skills to turn noisy measurements into conclusions you can defend