Senior/Staff ML Engineer, AI-Assisted Efficiency Optimization

• Implement AI-driven methods for GPU kernel optimizations to enhance program efficiency and performance leveraging tools such as CompilerGym, KernelLLM, etc.
• Develop strategies for resource-efficient deployment of intelligent optimization processes using frameworks such as AutoFDO.
• Guide high-level optimizations that include both highly optimized kernels, as well as AI-driven neural architecture search (NAS) to optimize training efficiency and inference latency.
• Assess performance improvements via evaluation metrics and real-world feedback.

This role contributes to making autonomy accessible to all by improving AI efficiency, which supports Nuro's mission of empowering a safer, richer, and more connected future through self-driving technology.

• 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, undertake complex assignments independently, and while enjoying to balance innovative exploration with practical considerations.

Base pay range between $193,930 and $352,290, annual performance bonus, equity, and a competitive benefits package. Nuro celebrates differences and is committed to a diverse workplace that fosters inclusion and psychological safety for all employees.

Nuro is a self-driving technology company on a mission to make autonomy accessible to all, building the world's most scalable driver by combining cutting-edge AI with automotive-grade hardware to support applications like robotaxis and commercial fleets.