Staff Machine Learning Engineer - Applied AI jobs in United States
info-icon
This job has closed.
company-logo

Uber · 12 hours ago

Staff Machine Learning Engineer - Applied AI

Uber is a technology company that builds intelligent systems to enhance product experiences for riders, drivers, merchants, and couriers. They are seeking a Staff Machine Learning Engineer to lead the development of AI products and ML infrastructure, ensuring production reliability and performance at scale while collaborating across various teams.

LogisticsMobile AppsRide SharingSoftwareTransportation
check
H1B Sponsor Likelynote

Responsibilities

Build end-to-end AI products - from prototype to scalable production deployment - integrating LLMs and multimodal AI into Uber's consumer, earner, and enterprise experiences
Implement automated evaluation systems that use LLM-as-a-judge techniques to benchmark model quality, ensure consistency, and accelerate experimentation
Design and implement high-throughput, low-latency backend services and APIs that connect to leading AI models (e.g., OpenAI, Claude, Gemini, Mistral), ensuring production reliability, low latency, fault tolerance, and cost optimization at scale
Lead the development of ML infrastructure for training, fine-tuning, evaluation, and deployment - including feature pipelines, model serving, offline/online consistency, and experiment management
Own production ML systems end-to-end, including rollout strategies, monitoring, alerting, capacity planning, and on-call readiness. Establish best practices for ML systems design, including versioning, reproducibility, data validation, model lifecycle management, and safe deployment
Collaborate across disciplines (engineering, product, design, and data science) to define user problems and translate them into AI-powered solutions
Mentor engineers and data scientists, fostering a culture of technical excellence and cross-functional learning

Qualification

Machine LearningDistributed SystemsML InfrastructureLarge Language ModelsPythonData Processing SystemsBackend ServicesCollaboration SkillsCommunication Skills

Required

10+ years of experience in software engineering, data science, or machine learning, including a track record of shipping production AI systems
Deep understanding of large language models, including fine-tuning, prompt engineering, embeddings, and retrieval-augmented generation (RAG)
Strong backend and distributed systems expertise, with experience designing and operating highly available, scalable services in production
Deep experience with ML infrastructure, including model training pipelines, online serving systems, feature stores, experiment platforms, and evaluation frameworks
Proficiency in Python, Go, or Java, with demonstrated ability to build data- and compute-intensive backend systems
Hands-on experience with distributed data processing systems (e.g., Spark, Flink, Ray) and workflow orchestration (e.g., Airflow or equivalent)
Ability to analyze data, run experiments, and derive insights for model and product improvement
Excellent communication and collaboration skills across technical and non-technical teams

Preferred

Master's or Ph.D. in Computer Science, Data Science, or related field
Experience integrating foundation model APIs (OpenAI, Claude, Gemini, Cohere, etc.) into production-grade systems
Proven ability to architect AI-powered backend services, optimizing for scalability, latency, and cost efficiency
Background in LLM evaluation systems or AI agent orchestration frameworks (LangChain, Semantic Kernel, etc.)
Demonstrated success leading cross-functional projects that deliver measurable user or business impact
Familiarity with multimodal AI (text, speech, and image models) and data-centric development workflows
Strong understanding of model serving architectures, including online inference, batch inference, caching strategies, and GPU utilization

Benefits

You will be eligible to participate in Uber's bonus program
May be offered an equity award & other types of comp
You will also be eligible for various benefits.

Company

Uber develops, markets, and operates a ride-sharing mobile application that allows consumers to submit a trip request.

H1B Sponsorship

Uber has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (830)
2024 (796)
2023 (684)
2022 (954)
2021 (750)
2020 (638)

Funding

Current Stage
Public Company
Total Funding
$35.56B
Key Investors
William AckmanPayPalToyota Motor
2025-09-08Post Ipo Debt· $2.25B
2025-05-13Post Ipo Debt· $1B
2025-01-01Post Ipo Equity· $2.3B

Leadership Team

leader-logo
Dara Khosrowshahi
CEO
linkedin
leader-logo
Prashanth Mahendra -Rajah
Chief Financial Officer
linkedin
Company data provided by crunchbase