Staff Software Engineer - ML Infra

Posted Yesterday
Be an Early Applicant
Sunnyvale, CA, USA
In-Office
232K-258K Annually
Senior level
Logistics • Transportation • 3PL: Third Party Logistics
We reimagine the way the world moves for the better.
The Role
Develop scalable ML infrastructure for autonomous vehicle use cases within Uber AV Labs. Responsibilities include designing multi-region, multi-cloud platforms; customizing Uber’s ML platform; aligning infrastructure and AV teams; executing technical projects from conception through production; and optimizing GPU efficiency. The role requires expertise in CUDA, TensorRT, Kubernetes, Flyte or Airflow, and Ray, along with at least six years of experience and a computer science degree.
Summary Generated by Built In
About the Role

Uber is launching AV Labs to accelerate the autonomous technology ecosystem. We’re building out a high-velocity team of multi-disciplinary experts to turn real-world operations into high-quality data for our autonomous partners. This team will be focused on the hardest problem in AV today: unlocking real-world, long-tail driving data. Autonomy is now a data race–and Uber has an edge: We collect rare, real-world driving data at a scale and capital efficiency no one else can match (millions of Uber trips every hour across cities, conditions, and edge cases create the data autonomy has been missing).

 

We will build platforms that harness scale and real-world complexity to reimagine how the world moves.

You will be a software engineer in AV Labs and involved in the development and implementation of the latest technology for self-driving stacks. The ideal candidate will be able to identify issues, provide solutions, and implement the fixes, as well as set a high technical excellence bar in all things we do.

What the Candidate Will Do:
  • Work with Uber’s ML platform and design customized solutions for AV use cases.
  • Design and implement scalable infrastructure on multiple regions and multiple cloud providers.
  • Lead stakeholder alignment with Uber infrastructure teams and AV Labs teams.
  • Drive efficiency efforts on GPUs.
Basic Qualifications:
  • Minimum 6 years of working experience.
  • Proven experience for executing technical projects from conception to production.
  • Bachelor’s degree in Computer Science or related fields.
  • Familiar with GPU technologies: CUDA, TensorRT, GPU optimizations
  • Familiar with compute cluster management: Kubernetes
  • Familiar with batch orchestration: Flyte or Airflow
  • Familiar with distributed batch processing: Ray
Preferred Qualifications
  • Master or PhD degree in Computer Science or related fields.
  • Background in ML Platform teams from large companies.
Responsibilities

For Sunnyvale, CA-based roles: The base salary range for this role is USD $232,000 per year - USD $258,000 per year.

You will be eligible to participate in Uber's bonus program, and may be offered an equity award & other types of comp. All full-time employees are eligible to participate in a 401(k) plan. You will also be eligible for various benefits.

About Us

Ready to Ride?

This isn't the kind of place where you follow a playbook — it's where you help write one. If you're driven by impact, energized by challenge, and ready to shape how the world moves — we'd love to hear from you.

You may be eligible for bonuses, equity, and other compensation, as well as a range of benefits. Explore our benefits.

Offices remain key to collaboration and Uber's culture. Unless approved for full remote work, employees must spend at least 50% of their time in-office. Some roles, like those at greenlight hubs, require full-time in-office presence. Ask your Recruiter for details about this role's requirements.

Uber is proud to be an Equal Opportunity employer. All qualified applicants will receive consideration for employment without regard to sex, gender identity, sexual orientation, race, color, religion, national origin, disability, protected Veteran status, age, or any other characteristic protected by law. We also consider qualified applicants regardless of criminal histories, consistent with legal requirements. If you have a disability or special need that requires accommodation, please let us know by completing this form.

Skills Required

  • Minimum 6 years of working experience
  • Experience executing technical projects from conception to production
  • Bachelor's degree in Computer Science or a related field
  • Experience with GPU technologies, including CUDA, TensorRT, and GPU optimization
  • Experience with Kubernetes for compute cluster management
  • Experience with Flyte or Airflow for batch orchestration
  • Experience with Ray for distributed batch processing
  • Master's or PhD degree in Computer Science or a related field
  • Background working on ML Platform teams at large companies

Uber Compensation & Benefits Highlights

The following summarizes recurring compensation and benefits themes identified from responses generated by popular LLMs to common candidate questions about Uber and has not been reviewed or approved by Uber.

  • Parental & Family Support Policies provide a minimum of fully paid parental leave for all parents and financial support for fertility, adoption, and surrogacy, with added credits to ease the transition. Programs extend to family medical leave and parenting support resources, indicating depth beyond baseline offerings.
  • Healthcare Strength Healthcare coverage is described as comprehensive across many countries, with medical, dental, vision, life, disability, and mental health benefits, plus allowances where direct plans are not available. Wellness programs and reimbursements further reinforce access to care.
  • Wellbeing & Lifestyle Benefits Monthly ride and meal credits, free office meals/snacks, fitness stipends, onsite gyms, and wellbeing reimbursements create meaningful everyday value. Home‑office stipends, travel medical coverage, and counseling support round out lifestyle-oriented perks.

Uber Insights

Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: San Francisco, CA
21,000 Employees
Year Founded: 2009

What We Do

We are Uber. The go-getters. The kind of people who are relentless about our mission to help people go anywhere and get anything. Movement is what we do. It’s our lifeblood. It runs through our veins. It’s what gets us out of bed each morning. It pushes us to constantly reimagine how we can move better. For you. For all the places you want to go. For all the things you want to get. For all the ways you want to earn. Across the entire world. In real-time. At the incredible speed of now.

Why Work With Us

We welcome people from all backgrounds who seek the opportunity to help build a future where everyone and everything can move independently. If you have the curiosity, passion, and collaborative spirit, work with us, and let’s move the world forward, together.

Gallery

Gallery

Similar Jobs

Tubi Logo Tubi

Staff Software Engineer

News + Entertainment
In-Office or Remote
3 Locations
504 Employees
227K-325K Annually

Enverus Logo Enverus

Test Automation Engineer

Big Data • Information Technology • Software • Analytics • Energy
In-Office or Remote
2 Locations
1800 Employees

MetLife Logo MetLife

Group Benefits Sales Intern - Irvine

Fintech • Information Technology • Insurance • Financial Services • Big Data Analytics
Hybrid
Irvine, CA, USA
43000 Employees
24-24 Hourly

Applied Systems Logo Applied Systems

Manager, HRIS

Artificial Intelligence • Cloud • Payments • Software • Business Intelligence • Generative AI • Automation
Remote or Hybrid
United States
3116 Employees
100K-160K Annually

Similar Companies Hiring

Toro TMS Thumbnail
Cloud • Enterprise Web • Sales • Software • Transportation
Chicago, IL
80 Employees
Axle Health Thumbnail
Artificial Intelligence • Healthtech • Information Technology • Logistics
Santa Monica, CA
25 Employees
Vega Thumbnail
Artificial Intelligence • Automotive • Insurance • Transportation
US
43 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account