Staff Machine Learning Engineer - Applied AI

Uber Uber · Consumer · Sunnyvale, CA +2 · Engineering

Staff ML Engineer to define and lead the foundation model strategy for AI-native discovery experiences across Uber's Mobility and Delivery platforms, focusing on Search, Recommendations, and Conversational AI. The role involves end-to-end technical strategy, architecture decisions, leading cross-team initiatives, defining investment areas (build vs fine-tune vs partner), and mentoring senior engineers.

What you'd actually do

  1. Own the end-to-end technical strategy for foundation models across Search, Recommendations, and Conversational AI.
  2. Drive architecture decisions that influence multiple product surfaces (Eats, Grocery, Retail, Mobility).
  3. Lead cross-team initiatives spanning Retrieval, Ranking, Personalization, and LLM-powered assistants.
  4. Define long-term investment areas (build vs fine-tune vs partner models).
  5. Mentor senior engineers and act as a technical multiplier across the org.

Skills

Required

  • Masters degree or Ph.D in Computer Science, Engineering, Mathematics
  • 8+ years of ML experience
  • large-scale deep learning systems
  • high-impact ML systems in search, recommendations, or conversational AI
  • transformers
  • retrieval systems
  • ranking
  • embedding architectures
  • PyTorch
  • distributed training
  • influencing technical direction

Nice to have

  • Experience leading multi-team ML initiatives
  • Defined long-term technical roadmaps adopted across orgs
  • Elevated engineering standards through mentorship and technical leadership

What the JD emphasized

  • large-scale deep learning systems
  • high-impact ML systems in search, recommendations, or conversational AI
  • Deep expertise in transformers, retrieval systems, ranking, and embedding architectures
  • Strong experience with PyTorch and distributed training
  • Track record of influencing technical direction across teams

Other signals

  • foundation models
  • large-scale deep learning systems
  • transformers
  • retrieval systems
  • ranking
  • embedding architectures
  • PyTorch
  • distributed training