Senior​/ML Engineer, AI-Assisted Efficiency Optimization

Senior/Staff ML Engineer, AI-Assisted Efficiency Optimization

Join Nuro as a Senior/Staff ML Engineer to lead AI‑assisted GPU kernel optimization projects that boost flop utilization and reduce inference latency.

Nuro is a self‑driving technology company on a mission to make autonomy accessible to all. Founded in 2016, Nuro builds the world’s most scalable driver by combining cutting‑edge AI with automotive‑grade hardware. Its core technology, the Nuro Driver™, is licensed to a wide range of applications from robotaxis and commercial fleets to personally owned vehicles.

We are looking for an experienced, goal‑oriented Machine Learning Engineer to oversee our GenAI‑assisted GPU kernel optimizations project delivery. This initiative aims to surpass human‑engineered performance by applying advanced AI technologies.

• Implement AI‑driven methods for GPU kernel optimizations to enhance program efficiency and performance, leveraging tools such as Compiler Gym and Kernel
  
  LLM.
• Develop strategies for resource‑efficient deployment of intelligent optimization processes using frameworks such as AutoFDO.
• Guide high‑level optimizations that include highly optimized kernels and AI‑driven neural architecture search (NAS) to improve training efficiency and inference latency.
• Assess performance improvements via evaluation metrics and real‑world feedback.
• Collaborate with internal teams to benchmark processes and strategies for future AI‑assisted optimizations.
• Utilize state‑of‑the‑art leader board scoring systems to test AI model‑generated GPU kernels.
• Stay current with industry advances and incorporate emerging technologies into optimization workflows.

• Bachelor’s or Master’s Degree in Computer Science, Engineering, or a related field.
• Extensive experience with AI models, including but not limited to LLMs.
• Strong understanding of retrieval‑augmented generation (RAG) and language models (LLMs).
• Ability to self‑motivate and undertake complex assignments independently while balancing innovative exploration with practical considerations.

• Experience with GPU programming, CUDA, Triton, and machine learning algorithms.
• Familiarity with optimization techniques such as neural architecture search (NAS).
• Hands‑on experience with optimization programs, such as Google’s AutoFDO.

Base pay range: $193,930–$352,290 (depending on experience, qualifications, location, and skills). The role also includes an annual performance bonus, equity, and a competitive benefits package.

At Nuro, we celebrate differences and are committed to a diverse workplace that fosters inclusion and psychological safety for all employees. Nuro is proud to be an equal opportunity employer and expressly prohibits any form of workplace discrimination based on race, color, religion, gender, sexual orientation, gender identity or expression, national origin, age, genetic information, disability, veteran status, or any other legally protected characteristics.