Senior/Staff ML Engineer, AI-Assisted Efficiency Optimization

Who you are

You have a strong background in machine learning engineering with a focus on GPU optimizations — your experience includes leading projects that leverage AI technologies to enhance performance and efficiency. You are goal-oriented and thrive in collaborative environments, guiding teams to achieve high-level optimizations. Your expertise in AI-driven methods and tools such as CompilerGym and KernelLLM sets you apart in the field. You understand the intricacies of neural architecture search and its application in optimizing training efficiency and inference latency. You are adept at assessing performance improvements through evaluation metrics and real-world feedback, ensuring that your contributions lead to tangible results.

Desirable

Experience with resource-efficient deployment strategies and a solid understanding of advanced AI technologies will be beneficial. Familiarity with frameworks like AutoFDO and a passion for pushing the boundaries of AI-assisted optimizations will enhance your fit for this role.

What you'll do

In this role, you will lead the GenAI-assisted GPU kernel optimizations project, focusing on improving maximum flop utilization and reducing inference latency. You will implement AI-driven methods to enhance program efficiency, utilizing tools such as CompilerGym and KernelLLM. Your responsibilities will include developing strategies for resource-efficient deployment of intelligent optimization processes, guiding high-level optimizations, and collaborating with internal teams to benchmark processes and strategies. You will assess performance improvements via evaluation metrics and real-world feedback, ensuring that the optimizations you implement surpass human-engineered performance. Your leadership will be crucial in driving the project to success, fostering a culture of innovation and excellence within the team.

What we offer

Nuro provides a dynamic work environment where you can contribute to groundbreaking technology in the self-driving sector. You will be part of a mission-driven team that values inclusion and psychological safety, ensuring that every employee feels empowered to share their ideas and contribute to the company's success. We offer competitive compensation and opportunities for professional growth, allowing you to expand your skills and advance your career in a rapidly evolving industry. Join us in making autonomy accessible to all and be part of a future where technology enhances everyday life.