Drimva

Uber

Staff Software Engineer - ML Infra

$232K–$258K

Sunnyvale, California · Staff · Full-time · Posted Sep 30, 2026

Experience
Minimum 6 years of working experience.
Education
Bachelor’s degree in Computer Science or related fields.

About the job

Develop scalable ML infrastructure and customized solutions for autonomous-vehicle use cases, including GPU efficiency and coordination across infrastructure and AV teams.

What you'll do

  • Work with Uber’s ML platform to design customized solutions for AV use cases.
  • Design and implement scalable infrastructure across multiple regions and cloud providers.
  • Align stakeholders across Uber infrastructure and AV Labs teams, and drive GPU efficiency efforts.

What they ask for

  • Proven experience executing technical projects from conception to production.
  • Bachelor’s degree in Computer Science or a related field.
  • Familiarity with GPU technologies, including CUDA, TensorRT, and GPU optimizations.
  • Familiarity with Kubernetes, Flyte or Airflow, and Ray.

Nice to have

  • Master’s or PhD degree in Computer Science or a related field.
  • Background in ML Platform teams at large companies.

Pay

USD $232,000 per year - USD $258,000 per year (base salary)

Tools

CUDA, TensorRT, Kubernetes, Flyte, Airflow, Ray

AI focus

Build ML infrastructure and customized solutions supporting self-driving stacks and autonomous-vehicle use cases.

Model-written summary · the original posting is authoritative

First seen Oct 15, 2026 · Last verified Nov 4, 2026 (UTC)