Senior AI Infra Architect for Scalable ML Training

Uber Technologies Inc.; San Francisco (CA); On-site; USD 232,000 - 258,000; Full time

Benefits offered by this job

Bonus program

Equity award

401(k) plan

Benefits

Job summary

Uber Technologies Inc. is seeking a Staff Software Engineer for AI Infra - Training to design and build end-to-end systems powering the ML training stack at scale.

You will mentor engineers, diagnose performance issues, and drive efficient, secure architectures across global data centers and clouds. You will collaborate with Product, Data Science, and Operations to align technical investments with business value, ensuring peak efficiency and reliability in a highly dynamic environment.

Qualifications

• 8+ years of professional experience building scalable, fault-tolerant products or platforms.
• Proficiency in Go, C/C++, Java, Python, or JavaScript and ability to define organization-wide standards.
• Proficiency in state-of-the-art deep learning technologies and distributed training frameworks (LLMs, GPU optimizations).
• Proven experience leading large-scale engineering projects through inception to production.
• Ability to trade off buy vs. build and incremental vs. rewrite decisions.
• Experience identifying architectural gaps and reducing tech debt across the org.
• Bachelor's degree in Computer Science, Engineering, Mathematics, or equivalent.

Responsibilities

• Design and build end-to-end systems for Uber's ML training stack.
• Provide deep technical mentorship across the stack.
• Diagnose performance regressions and optimize resource utilization.
• Create tools and frameworks to boost velocity and quality across teams.
• Lead and mentor engineers, review code and designs.
• Collaborate cross-functionally with Product, Data Science, and Operations globally.

Skills

Go

C/C++

Java

Python

JavaScript

Deep learning

Distributed systems

Leadership

Education

Bachelor's degree in CS/Engineering

Tools

Distributed training frameworks

LLMs

GPU optimizations

Job description

Uber Technologies Inc. is seeking a Staff Software Engineer for AI Infra - Training to design and build end-to-end systems powering the ML training stack at scale.

You will mentor engineers, diagnose performance issues, and drive efficient, secure architectures across global data centers and clouds. You will collaborate with Product, Data Science, and Operations to align technical investments with business value, ensuring peak efficiency and reliability in a highly dynamic environment.