Staff Machine Learning Engineer

Uber Uber · Consumer · New York, NY +2 · Engineer - Software Engineering - Machine Learning & Artificial Intelligence

Uber is seeking a Staff Machine Learning Engineer to develop and productionize ML models and systems, build scalable ML infrastructure, and collaborate with product and research teams. This role focuses on the inference and serving aspects of ML models within Uber's diverse product ecosystem.

What you'd actually do

  1. Design, build, and maintain scalable and reliable ML infrastructure and systems
  2. Develop and productionize ML models and algorithms for various Uber products
  3. Collaborate with product managers, researchers, and other engineers to define and deliver ML solutions
  4. Optimize ML model performance for latency, throughput, and cost
  5. Mentor junior engineers and contribute to the team's technical growth

Skills

Required

  • Experience with ML frameworks (e.g., TensorFlow, PyTorch)
  • Proficiency in Python and/or Java/Scala
  • Experience with distributed systems and cloud platforms (e.g., AWS, GCP, Azure)
  • Strong understanding of ML model serving and inference optimization
  • Experience with MLOps best practices

Nice to have

  • Experience with large-scale data processing (e.g., Spark, Flink)
  • Familiarity with containerization technologies (e.g., Docker, Kubernetes)
  • Experience with A/B testing and experimentation

Other signals

  • Develop and productionize ML models and systems
  • Build scalable and reliable ML infrastructure
  • Collaborate with product and research teams