Senior Staff Machine Learning Engineer – Moonshot AI

Uber Uber · Consumer · Sunnyvale, CA · Engineering

Senior Staff ML Engineer for Uber's Moonshot AI team, focusing on building an enterprise AI data platform. Responsibilities include marketplace optimization, custom model development for annotation workflows (audio, video, text), automated quality evaluation using GenAI/LLM-as-Judge, and ML research. The role involves end-to-end ownership from research to production, technical leadership, and mentoring.

What you'd actually do

  1. Shape the technical vision and roadmap for Moonshot AI's ML initiatives, identifying strategic investments that advance both technical excellence and business objectives
  2. Architect foundational ML platforms and systems for marketplace optimization and annotation automation—designing scalable, reliable solutions that serve as the technical foundation for multiple product areas
  3. Drive end-to-end ML solutions from conception through production deployment, owning critical technical decisions on architecture, tooling, and infrastructure that impact millions of tasks and revenue growth
  4. Lead GenAI innovation : design and implement cutting-edge systems using custom SLMs, computer vision, and LLMs to provide ML assistance for annotations across audio, video, and text workflows while establishing quality standards
  5. Advance AI research capabilities : establish research direction, design benchmarks, contribute to research and publications that positions Uber AI Solutions as a thought leader

Skills

Required

  • PyTorch
  • TensorFlow
  • JAX
  • Computer Vision
  • Natural Language Processing
  • Deep Learning
  • Generative AI
  • distributed training infrastructure
  • large-scale model development
  • ML platform design

Nice to have

  • Ph.D. in Computer Science, Machine Learning, Statistics, or related field with focus on ML research
  • Publications at top-tier AI/ML conferences
  • prompt engineering
  • RAG systems
  • multi-task learning
  • model serving
  • feature stores
  • experiment platforms
  • observability systems
  • marketplace optimization
  • recommendation systems
  • multi-armed bandits
  • anomaly detection

What the JD emphasized

  • 10+ years of industry experience developing and shipping production machine learning models
  • Proven track record of technical leadership on large-scale ML initiatives with measurable business impact
  • Deep expertise across multiple areas: Computer Vision, Natural Language Processing, Deep Learning, and Generative AI
  • Extensive experience with distributed training infrastructure, large-scale model development, and ML platform design
  • Expert-level experience with LLM fine-tuning techniques , prompt engineering, RAG systems, and multi-task learning

Other signals

  • enterprise AI data platform
  • AI-assisted data annotation and validation
  • custom SLMs, computer vision, and LLMs
  • LLM-as-Judge frameworks
  • automated quality evaluation