Staff Machine Learning Engineer

Uber Uber · Consumer · Bangalore, India · Engineering

Uber's Applied AI team is seeking a Staff ML Engineer to design, implement, and scale high-impact AI solutions, focusing on Generative AI, Computer Vision, and Personalization. The role involves leading technical projects, influencing ML system architecture, and collaborating cross-functionally to deliver AI-powered features from ideation to production.

What you'd actually do

  1. Design and implement ML-driven systems that power core Uber experiences, with a focus on scalability, reliability, and performance.
  2. Lead the technical execution of key projects involving classical ML, deep learning, and generative AI technologies (e.g., LLMs, multimodal models).
  3. Collaborate closely with product, data science, and infrastructure teams to develop AI solutions from ideation through production deployment.
  4. Contribute to and influence the technical direction for Applied AI, particularly around system design, model architecture, and infrastructure decisions.
  5. Champion engineering best practices in ML development — including experimentation workflows, model versioning, evaluation, monitoring, and responsible AI.

Skills

Required

  • 10+ years of industry experience in machine learning or software engineering
  • Strong knowledge of machine learning, deep learning, and exposure to generative AI techniques (e.g., transformers, LLMs, diffusion).
  • Experience designing and scaling ML systems or platforms, including training pipelines, serving infrastructure, and model lifecycle tooling.
  • Fluency in ML frameworks (e.g., PyTorch, TensorFlow, JAX) and development in Python and/or scalable backend languages (e.g., Java, Go).
  • Excellent collaboration and communication skills

What the JD emphasized

  • proven record of delivering ML solutions to production
  • designing and scaling ML systems or platforms
  • Generative AI
  • Computer Vision
  • Personalization

Other signals

  • delivering ML solutions to production
  • scaling ML systems or platforms
  • Generative AI
  • Computer Vision
  • Personalization