Skip to main contentA logo with &quat;the muse&quat; in dark blue text.

Staff ML Engineer - Applied AI

AT Uber
Uber

Staff ML Engineer - Applied AI

Bangalore, India

About the Role

Applied AI is a horizontal AI team at Uber partnering with product and platform teams across the company to deliver cutting-edge machine learning solutions for core business problems. We specialize in areas like Generative AI, Computer Vision, Personalization, and the ML infrastructure needed to scale these systems in production.

We're looking for a Staff Engineer with deep experience in classical machine learning, generative AI, and ML systems design to help build and scale high-impact AI solutions. In this role, you'll lead technically complex projects, influence the architecture of ML systems, and collaborate cross-functionally to drive innovation and impact across multiple Uber surfaces.

Want more jobs like this?

Get Data and Analytics jobs in Bangalore, India delivered to your inbox every week.

By signing up, you agree to our Terms of Service & Privacy Policy.

This is a great opportunity for a strong technical leader who thrives in a fast-paced, product-driven environment and wants to be at the forefront of applied AI at scale.

What You'll Do

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

Basic Qualifications

  • 10+ years of industry experience in machine learning or software engineering, with a proven record of delivering ML solutions to production.
  • 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 with the ability to work across teams and functions.

Preferred Qualifications

  • PhD in Computer Science, Machine Learning, or a related field.
  • Hands-on experience integrating LLMs or generative models into product experiences (e.g., summarization, personalization, automation).
  • Familiarity with MLOps, experimentation frameworks, or ML observability tools.
  • Track record of technical leadership in multi-disciplinary projects involving engineering, data science, and product.

We welcome people from all backgrounds who seek the opportunity to help build a future where everyone and everything can move independently. If you have the curiosity, passion, and collaborative spirit, work with us, and let's move the world forward, together.

Offices continue to be central to collaboration and Uber's cultural identity. Unless formally approved to work fully remotely, Uber expects employees to spend at least half of their work time in their assigned office. For certain roles, such as those based at green-light hubs, employees are expected to be in-office for 100% of their time. Please speak with your recruiter to better understand in-office expectations for this role.

*Accommodations may be available based on religious and/or medical conditions, or as required by applicable law. To request an accommodation, please reach out to accommodations@uber.com.

Client-provided location(s): Bengaluru, Karnataka, India
Job ID: Uber-142400
Employment Type: Full Time

Perks and Benefits

  • Health and Wellness

    • Health Insurance
    • Health Reimbursement Account
    • Dental Insurance
    • Vision Insurance
    • Life Insurance
    • FSA With Employer Contribution
    • Fitness Subsidies
    • On-Site Gym
    • Mental Health Benefits
  • Parental Benefits

    • Fertility Benefits
  • Work Flexibility

    • Flexible Work Hours
    • Remote Work Opportunities
    • Hybrid Work Opportunities
  • Office Life and Perks

    • Casual Dress
    • Pet-friendly Office
    • Snacks
    • Some Meals Provided
    • On-Site Cafeteria
  • Vacation and Time Off

    • Paid Vacation
    • Unlimited Paid Time Off
    • Paid Holidays
    • Personal/Sick Days
    • Sabbatical
    • Volunteer Time Off
  • Financial and Retirement

    • 401(K)
    • Company Equity
    • Performance Bonus
  • Professional Development

    • Work Visa Sponsorship
    • Associate or Rotational Training Program
    • Promote From Within
    • Mentor Program
    • Access to Online Courses
  • Diversity and Inclusion

    • Employee Resource Groups (ERG)
    • Diversity, Equity, and Inclusion Program