Nuro · 2 months ago
Senior/Staff ML Engineer, AI-Assisted Efficiency Optimization
Mountain View, CA
Full-time
Onsite
Senior Level, Lead/Staff
$194K/yr - $352K/yrNuro is a self-driving technology company dedicated to making autonomy accessible to all. They are seeking an experienced Machine Learning Engineer to lead GPU kernel optimizations, focusing on improving performance and reducing latency through advanced AI technologies.
Artificial Intelligence (AI)Autonomous VehiclesRoboticsTransportation
Responsibilities
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
Collaborate with internal teams to benchmark processes and strategies for future AI-assisted optimizations
Utilize state-of-the-art leaderboard scoring systems to test the ability of different AI models to generate efficient GPU kernels
Stay current with industry advances
Qualification
Required
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
Preferred
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
Benefits
Annual performance bonus
Equity
Competitive benefits package
Company
Nuro
Nuro is a robotics company specializing in the development of autonomous driving technologies.
Founded in 2016
501-1000 employees
H1B Sponsorship
Nuro has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (50)
2024 (67)
2023 (91)
2022 (91)
2021 (68)
2020 (55)
Funding
Current Stage
Late StageTotal Funding
$2.34BKey Investors
Disney AcceleratorTiger Global ManagementT. Rowe Price
2025-08-21Series E· $97M
2025-04-09Series E· $106M
2024-04-06Secondary Market
Leadership Team
Jiajun Zhu
Co-Founder and Co-CEO
Dave Ferguson
Co-Founder
Recent News
Business Insider
2026-01-07
2026-01-07
Business – Baltimore Sun
2026-01-07
Company data provided by crunchbase