Staff Software Engineer - ML Infra

Job description

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.