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 autonomous driving technology.
Key Responsibilities
- Implement AI-driven GPU kernel optimization methods to improve program efficiency and performance.
- Develop strategies for resource-efficient deployment of AI-based optimization processes.
- Guide high-level optimizations, including neural architecture search, to enhance training efficiency and inference latency.
- Assess performance improvements using evaluation metrics and real-world feedback.
- Collaborate with internal teams to benchmark and refine AI-assisted optimization strategies.
- Utilize leaderboard scoring systems to evaluate AI model effectiveness in generating 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
Jobs in Mountain View, California
More jobs at Nuro
Senior Software Engineer, Routing
Nuro
NEW
Mountain View
Full Time
13h
$176k-265k
Shift Supervisor, On-Road Operations
Nuro
NEW
Santa Clara
Full Time
13h
$80k-108k
Virtual Hiring Event Autonomous Vehicle Operator 12-Month Duration
Nuro
Santa Clara
TEMPORARY
5d
$0k-0k/hr
More jobs in Mountain View, California
Senior Audio systems engineer
Gatik
NEW
Mountain View
Full Time
13h
$170k-230k
AI Operations RAG Architect
Gatik
NEW
Mountain View
Full Time
13h
$170k-220k
IT IAM SaaS Integration Engineer
Gatik
NEW
Mountain View
Full Time
13h
$150k-210k