Staff Software Engineer - ML Infra

Job Description

Develop and implement technology for self-driving stacks, adapting Uber’s machine learning platform to autonomous vehicle use cases. Design scalable infrastructure across regions and cloud providers, align infrastructure and AV Labs stakeholders, and improve GPU efficiency.

Requirements: Requires at least six years of work experience, a bachelor’s degree in computer science or a related field, and a record of taking technical projects from conception to production. Candidates should be familiar with CUDA, TensorRT, GPU optimization, Kubernetes, Flyte or Airflow, and Ray; a master’s or PhD and experience on a large-company ML platform team are preferred.

Key Skills: Machine Learning Platform, Scalable Infrastructure, Multi-Region Systems, Multi-Cloud Systems, GPU Optimization, CUDA, TensorRT, Kubernetes, Flyte, Airflow, Ray, Distributed Batch Processing, Batch Orchestration, Technical Project Execution, Stakeholder Alignment

Benefits: Bonus Program, Equity Award, 401(k) Plan, Other Benefits