Staff Software Engineer - AI Platform (Michelangelo)

---- Basic Qualifications ----

1. Master in relevant fields (CS, EE, Math, Stats, etc.) AND 6-years full-time Software Engineering work experience in deep learning
2. Proficiency in Python and PyTorch
3. Expertise in designing, debugging, and optimizing distributed deep learning systems.
4. Working experience of distributed training in PyTorch at Scale (e.g., data parallelism, model parallelism).
5. Strong ability to translate complex DL requirements and problems into scalable solutions.

---- Preferred Qualifications ----

1. Expertise in distributed training frameworks such as DDP, DeepSpeed, FSDP, or TorchRec.
2. Familiarity with C++, Go or CUDA programming.
3. Expertise in optimizing GPU/TPU training performance and data loading efficiency.
4. Familiarity with large-scale distributed infrastructure tools like Ray, OpenAI Triton, PyTorch Lightning.
5. Built and deployed end-to-end machine learning systems in production.
6. Experience training large models (10B+ parameters), such as large recommendation systems or large language models (LLMs)
7. PhD in relevant fields (CS, EE, Math, Stats, etc.)

For San Francisco, CA-based roles: The base salary range for this role is USD$223,000 per year - USD$248,000 per year.

For Seattle, WA-based roles: The base salary range for this role is USD$223,000 per year - USD$248,000 per year.

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

For all US locations, you will be eligible to participate in Uber's bonus program, and may be offered an equity award & other types of comp. You will also be eligible for various benefits. More details can be found at the following link https://www.uber.com/careers/benefits.

Uber's mission is to reimagine the way the world moves for the better. Here, bold ideas create real-world impact, challenges drive growth, and speed fuels progress. What moves us, moves the world - let's move it forward, together.

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.

Offices continue to be central to collaboration and Uber's cultural identity. Unless formally approved to work fully remotely, Uber expects employees to spend at least half of their work time in their assigned office. For certain roles, such as those based at green-light hubs, employees are expected to be in-office for 100% of their time. Please speak with your recruiter to better understand in-office expectations for this role.

Client-provided location(s): Seattle, WA, San Francisco, CA, Sunnyvale, CA

Job ID: Uber-143014

Employment Type: FULL_TIME

Posted: 2025-07-11T00:32:37