Principal AI Architect: Foundation Models & Large-Scale ML

Uber

San Francisco (CA)

On-site

USD 302,000 - 336,000

Full time

14 days+

A complete application in a minute

Tailored resume and cover letter, ready to send.

Get past ATS filters

Benefits offered by this job

401(k) plan
Bonus program
Equity award

Job summary

Uber is seeking a Principal Engineer, Machine Learning / AI to lead the development of large-scale Foundation Models trained on Uber’s data. This hands-on technical leadership role shapes model architectures, training strategies, and scaling across multiple engineering and product teams.

You will drive representation learning, ML impact, and scalable infrastructure while guiding architectural decisions and best practices.

Qualifications

  • BS degree in Computer Science or related field, or equivalent experience.
  • 10+ years of software engineering experience with proven technical leadership and innovation.
  • Extensive experience designing, building, and scaling deep learning systems and large-scale AI infrastructure.
  • MS/PhD in Computer Science or related field is a plus.
  • Hands-on experience with Foundation Models, pretraining, and distributed training architectures.

Responsibilities

  • Architect and build large-scale Foundation Models using proprietary data.
  • Define the technical direction for architecture, training approaches, and scaling.
  • Advance representation learning with embeddings for downstream ML models.
  • Drive company-wide ML impact by partnering across Mobility, Delivery, Earner, Merchant, and Consumer experiences.
  • Build scalable ML capabilities including training infrastructure and distributed compute.
  • Provide deep technical leadership and establish best practices for large-scale model development.
  • Collaborate with senior leaders, researchers, and engineers to align initiatives.
  • Remain hands-on and contribute to critical decisions and model development.

Skills

Technical leadership
Deep learning
Distributed computing
Architectural design
Hands-on leadership

Education

BS in CS or related field
MS/PhD preferred

Job description

Uber is seeking a Principal Engineer, Machine Learning / AI to lead the development of large-scale Foundation Models trained on Uber’s data. This hands-on technical leadership role shapes model architectures, training strategies, and scaling across multiple engineering and product teams.

You will drive representation learning, ML impact, and scalable infrastructure while guiding architectural decisions and best practices.

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

Principal ML Architect – Foundation Models
Principal ML Architect – Foundation Models

Uber • United States

On-site
USD 302,000 - 336,000
Bonus program and equity
401(k) plan
Benefits
Principal Engineer - ML / AI – Foundation Models
Principal Engineer - ML / AI – Foundation Models

Uber • San Francisco (CA)

On-site
USD 302,000 - 336,000
401(k) plan
Bonus program
Equity award
Principal Engineer - ML / AI - Foundation Models
Principal Engineer - ML / AI - Foundation Models

Uber • United States

On-site
USD 302,000 - 336,000
Bonus program and equity
401(k) plan
Benefits
Principal AI Infrastructure Architect for Foundation Models
Principal AI Infrastructure Architect for Foundation Models

Amazon Web Services (AWS) • Seattle (WA)

On-site
USD 183,000 - 247,000
Health insurance
401(k) matching
Paid time off
+1
Staff ML Engineer: Global-Scale Foundation Models
Staff ML Engineer: Global-Scale Foundation Models

Uber • Sunnyvale (CA)

On-site
USD 232,000 - 258,000
Bonus program
Equity award
401(k) plan
+1
Senior AI Infra Architect for Scalable ML Training
Senior AI Infra Architect for Scalable ML Training

Uber Technologies Inc. • San Francisco (CA)

On-site
USD 232,000 - 258,000
Bonus program
Equity award
401(k) plan
+1
Senior Staff ML Engineer: Personalization & Large-Scale AI
Senior Staff ML Engineer: Personalization & Large-Scale AI

Uber Technologies Inc. • San Francisco (CA)

On-site
USD 267,000 - 297,000
Bonus program
Equity award
401(k) plan
+1
Sr Staff ML Engineer - Applied AI
Sr Staff ML Engineer - Applied AI

Uber • United States

On-site
USD 267,000 - 297,000
Bonus program
Equity award
401(k) plan
Principal ML Engineer & AI Systems Architect
Principal ML Engineer & AI Systems Architect

Provectus • Michigan

On-site
USD 120,000 - 160,000
Health, dental, and vision insurance
401(k) with company match
Generous PTO
+1
Senior AI Engineer, Foundation Models & LLMs
Senior AI Engineer, Foundation Models & LLMs

Capital One • California

Hybrid
USD 251,000 - 286,000