Sr Software Engineer - ML Infrastructure

Posted Yesterday
Be an Early Applicant
4 Locations
In-Office
202K-224K Annually
Senior level
Logistics • Transportation • 3PL: Third Party Logistics
We reimagine the way the world moves for the better.
The Role
Build scalable ML infrastructure for model training, simulation, optimization, evaluation, and productionization. Develop shared frameworks, distributed execution systems, experimentation tools, orchestration, and deployment patterns across CPU and GPU environments. Improve performance, reliability, reproducibility, and compute efficiency while partnering with scientists and platform teams. Own systems through design, rollout, operations, incident response, and mentoring.
Summary Generated by Built In
Senior Software Engineer - ML Infrastructure, Mobility Simulation and PlanningAbout the role and team

How can Uber plan pricing and incentives to meet financial goals while keeping its marketplace healthy and growing? That’s the challenge our Mobility Simulation and Planning (MSP) organization tackles within Marketplace.

MSP builds forecasting, simulation, and optimization products that help Operations teams translate financial targets into pricing and incentive policies across hundreds of cities. Our planning product, Planner, brings these capabilities together to generate actionable price plans. Supporting this work requires scalable model training and simulation, efficient distributed computing, and reliable workflows for generating and deploying plans.

The MSP Infra team builds the data and computing foundations that make these capabilities possible. We partner with engineers, scientists, and shared-platform teams to turn complex modeling workflows into reliable, scalable, reusable infrastructure.

We’re looking for a Senior Software Engineer focused on ML infrastructure. You’ll build the frameworks, distributed execution capabilities, and developer tooling that help scientists develop, train, evaluate, and productionize models. Your work will make it easier to move from an experimental model to a dependable production workflow and to run the simulations and backtests needed to evaluate marketplace decisions.

How do we scale training and simulation across CPUs and GPUs? How do we make experiments reproducible and performance bottlenecks easy to diagnose? How do we support new modeling approaches without creating a separate infrastructure stack for every use case? You’ll solve these problems through hands-on engineering, close collaboration with modeling teams, and end-to-end ownership of production systems.

What you’ll do
  • Build shared ML frameworks: Design and evolve reusable interfaces and libraries for model development, training, inference, evaluation, and backtesting.
  • Scale distributed workloads: Build and optimize execution capabilities for training, simulation, and optimization across CPU and GPU infrastructure.
  • Accelerate experimentation: Develop tooling for hyperparameter tuning, experiment tracking, reproducibility, and performance analysis so scientists can evaluate ideas faster.
  • Simplify productionization: Create reliable orchestration, configuration, dependency management, and deployment patterns that reduce the gap between experimentation and production.
  • Improve performance and efficiency: Profile workloads, diagnose compute and memory bottlenecks, and make thoughtful tradeoffs across runtime, resource utilization, reliability, and cost.
  • Build on shared platforms: Partner with Uber’s ML and compute platform teams to adopt and extend common capabilities rather than maintain isolated infrastructure.
  • Own systems and raise the engineering bar: Lead projects from design through rollout and operations, participate in on-call and incident response, and mentor engineers through design and code reviews.
Basic Qualifications
  • 6+ years of professional software engineering experience, including work on ML infrastructure, distributed computing, or production platforms for computationally intensive workloads.
  • Strong programming skills in Python and experience building maintainable, well-tested software.
  • Experience designing, building, and operating distributed systems or workflow execution infrastructure.
  • Experience supporting production ML workflows, such as model training, batch inference, evaluation, or deployment.
  • Understanding of performance fundamentals, including parallelism, memory management, data movement, and resource utilization.
  • Experience delivering technical projects from design through production rollout and ongoing operations.
  • Ability to communicate technical tradeoffs and collaborate across engineering, data science, and platform teams.
  • Bachelor’s or Master’s degree in Computer Science, Engineering, or equivalent practical experience.
Preferred Qualifications
  • Experience with Ray or comparable distributed computing frameworks.
  • Familiarity with ML frameworks such as PyTorch, PyTorch Lightning, or LightGBM.
  • Experience with distributed training, GPU performance profiling, or CPU/GPU workload optimization.
  • Experience building hyperparameter tuning, experiment tracking, or model evaluation infrastructure.
  • Experience supporting large-scale simulation, backtesting, forecasting, or optimization workloads.
  • Familiarity with workflow orchestration, containerized execution, CI/CD, and build or dependency-management systems.
  • Experience designing shared frameworks or developer platforms adopted by multiple modeling teams.
  • A track record of measurable improvements in training or inference performance, infrastructure reliability, compute efficiency, or model-development productivity.

Responsibilities

For New York City, NY-based roles: The base salary range for this role is USD $202,000 per year - USD $224,000 per year.


For San Francisco, CA-based roles: The base salary range for this role is USD $202,000 per year - USD $224,000 per year.


For Seattle, WA-based roles: The base salary range for this role is USD $202,000 per year - USD $224,000 per year.


For Sunnyvale, CA-based roles: The base salary range for this role is USD $202,000 per year - USD $224,000 per year.


For all US locations, you will be eligible to participate in Uber's bonus program, and may be offered an equity award & other types of comp. All full-time employees are eligible to participate in a 401(k) plan. You will also be eligible for various benefits.

About Us

Ready to Ride?

This isn't the kind of place where you follow a playbook — it's where you help write one. If you're driven by impact, energized by challenge, and ready to shape how the world moves — we'd love to hear from you.

You may be eligible for bonuses, equity, and other compensation, as well as a range of benefits. Explore our benefits.

Offices remain key to collaboration and Uber's culture. Unless approved for full remote work, employees must spend at least 50% of their time in-office. Some roles, like those at greenlight hubs, require full-time in-office presence. Ask your Recruiter for details about this role's requirements.

Uber is proud to be an Equal Opportunity employer. All qualified applicants will receive consideration for employment without regard to sex, gender identity, sexual orientation, race, color, religion, national origin, disability, protected Veteran status, age, or any other characteristic protected by law. We also consider qualified applicants regardless of criminal histories, consistent with legal requirements. If you have a disability or special need that requires accommodation, please let us know by completing this form.

Skills Required

  • 6+ years of professional software engineering experience, including ML infrastructure, distributed computing, or production platforms for computationally intensive workloads
  • Strong programming skills in Python and experience building maintainable, well-tested software
  • Experience designing, building, and operating distributed systems or workflow execution infrastructure
  • Experience supporting production ML workflows, such as model training, batch inference, evaluation, or deployment
  • Understanding of performance fundamentals, including parallelism, memory management, data movement, and resource utilization
  • Experience delivering technical projects from design through production rollout and ongoing operations
  • Ability to communicate technical tradeoffs and collaborate across engineering, data science, and platform teams
  • Bachelor's or Master's degree in Computer Science, Engineering, or equivalent practical experience
  • Experience with Ray or comparable distributed computing frameworks
  • Familiarity with ML frameworks such as PyTorch, PyTorch Lightning, or LightGBM
  • Experience with distributed training, GPU performance profiling, or CPU/GPU workload optimization
  • Experience building hyperparameter tuning, experiment tracking, or model evaluation infrastructure
  • Experience supporting large-scale simulation, backtesting, forecasting, or optimization workloads
  • Familiarity with workflow orchestration, containerized execution, CI/CD, and build or dependency-management systems
  • Experience designing shared frameworks or developer platforms adopted by multiple modeling teams
  • Track record of measurable improvements in training or inference performance, infrastructure reliability, compute efficiency, or model-development productivity

Uber Compensation & Benefits Highlights

The following summarizes recurring compensation and benefits themes identified from responses generated by popular LLMs to common candidate questions about Uber and has not been reviewed or approved by Uber.

  • Parental & Family Support — Policies provide a minimum of fully paid parental leave for all parents and financial support for fertility, adoption, and surrogacy, with added credits to ease the transition. Programs extend to family medical leave and parenting support resources, indicating depth beyond baseline offerings.
  • Healthcare Strength — Healthcare coverage is described as comprehensive across many countries, with medical, dental, vision, life, disability, and mental health benefits, plus allowances where direct plans are not available. Wellness programs and reimbursements further reinforce access to care.
  • Wellbeing & Lifestyle Benefits — Monthly ride and meal credits, free office meals/snacks, fitness stipends, onsite gyms, and wellbeing reimbursements create meaningful everyday value. Home‑office stipends, travel medical coverage, and counseling support round out lifestyle-oriented perks.

Uber Insights

Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: San Francisco, CA
21,000 Employees
Year Founded: 2009

What We Do

We are Uber. The go-getters. The kind of people who are relentless about our mission to help people go anywhere and get anything. Movement is what we do. It’s our lifeblood. It runs through our veins. It’s what gets us out of bed each morning. It pushes us to constantly reimagine how we can move better. For you. For all the places you want to go. For all the things you want to get. For all the ways you want to earn. Across the entire world. In real-time. At the incredible speed of now.

Why Work With Us

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.

Gallery

Gallery

Similar Jobs

Philo Logo Philo

Software Engineer

Cloud • Digital Media • News + Entertainment • On-Demand
Remote or Hybrid
2 Locations
165 Employees
165K-250K Annually

Enverus Logo Enverus

Account Director

Big Data • Information Technology • Software • Analytics • Energy
In-Office or Remote
5 Locations
1800 Employees

Wells Fargo Logo Wells Fargo

Product Manager

Fintech • Financial Services
Hybrid
New York, NY, USA
205000 Employees
215K-355K Annually

MetLife Logo MetLife

ADA Unit Leader

Fintech • Information Technology • Insurance • Financial Services • Big Data Analytics
Remote or Hybrid
United States
43000 Employees
59K-99K Annually

Similar Companies Hiring

Toro TMS Thumbnail
Cloud • Enterprise Web • Sales • Software • Transportation
Chicago, IL
80 Employees
Axle Health Thumbnail
Artificial Intelligence • Healthtech • Information Technology • Logistics
Santa Monica, CA
25 Employees
Vega Thumbnail
Artificial Intelligence • Automotive • Insurance • Transportation
US
65 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account