Job Description
A Machine Learning Engineer role focused on developing AI-driven GPU kernel optimizations to improve performance and reduce inference latency, supporting Nuro's self-driving technology platform.
Key Responsibilities
- Implement AI-driven methods for GPU kernel optimizations to improve efficiency and performance.
- Develop strategies for resource-efficient deployment of AI optimization processes using frameworks like AutoFDO.
- Guide high-level optimizations, including neural architecture search (NAS), to enhance training efficiency and inference latency.
- Assess performance improvements through evaluation metrics and real-world feedback.
- Collaborate with internal teams to benchmark optimization processes and strategies.
- Utilize leaderboard scoring systems to test AI models' ability to generate efficient GPU kernels.
Requirements
- Bachelor's or Master's Degree in Computer Science, Engineering, or a related field.
- Extensive experience with AI models, including but not limited to Large Language Models (LLMs).
- Strong understanding of retrieval-augmented generation (RAG) and Language Models (LLMs).
- Ability to self-motivate, undertake complex assignments independently, and balance innovative exploration with practical considerations.
Benefits & Perks
Compensation/salary range between $193,930 and $352,290 depending on experience and qualifications
Annual performance bonus
Equity
Competitive benefits package
Ready to Apply?
Join Nuro and make an impact in renewable energy
Stay Updated on Sustainability Jobs
Get the latest renewable energy jobs and career tips delivered to your inbox.
Job Alerts
Get notified about new sustainability jobs
More at Nuro
More jobs at Nuro
Embedded Software Engineer Intern
Nuro
NEW
Mountain View
Full Time
13h
Senior Project Manager, Software Operations
Nuro
NEW
Mountain View
Full Time
2d
$132k-198k
Safety Governance Standards Lead
Nuro
NEW
Mountain View
Full Time
2d
$194k-291k
More jobs in Mountain View, California
Staff Software Engineer - Data Curation
Aurora
Mountain View
Full Time
Nov 25
$189k-303k
Senior Software Engineer, Visualization
Aurora
Mountain View
Full Time
Dec 24
$139k-223k
Senior Software Engineer, Behavior Simulation
Aurora
Mountain View
Full Time
Oct 30
$139k-223k