Staff Machine Learning Engineer

The Staff Machine Learning Engineer role is focused on modernizing the Uber Freight marketplace team. In this onsite position in Sunnyvale, CA, you will own end-to-end machine learning models for predictive modeling and marketplace optimization, collaborating closely with backend engineers to productionize models and maintain performance.

Role Overview

Work within the Uber Freight Marketplace team to develop and operationalize ML solutions that support key marketplace capabilities, including cost prediction, booking probability, and demand elasticity. You will maintain ownership across the lifecycle of the models and ensure they perform as intended in production environments.

Responsibilities

• Provide end-to-end ownership of machine learning models for the UF marketplace, including cost prediction, booking probability, and demand elasticity.
• Partner with backend engineers to implement models in production and verify that the deployed systems behave as expected.

Required Qualifications

• 6+ years of experience developing machine learning models to solve business problems.
• Bachelor’s degree in Computer Science, Computer Engineering, or a related field.
• Familiarity with modern AI/ML frameworks, for example PyTorch.

Preferred Qualifications

• Product experience is a strong plus.
• Ability to adapt machine learning models to the business context is critical.
• Prior experience with state-of-the-art marketplace technology is preferred.
• Experience with causal inference and constrained optimization.

Technology Stack

• PyTorch

Location and Compensation

Location: Sunnyvale, CA (onsite).
Salary: USD 232,000 - 258,000 per year.

Benefits

• Eligible to participate in Uber’s bonus program (for all US locations).
• May be offered an equity award and other types of compensation (for all US locations).
• All full-time employees are eligible to participate in a 401(k) plan.
• Eligible for various benefits.

Base Salary Notice by Location

• Chicago, IL: USD 209,000 - 232,000 per year.
• New York City, NY: USD 232,000 - 258,000 per year.
• San Francisco, CA: USD 232,000 - 258,000 per year.
• Seattle, WA: USD 232,000 - 258,000 per year.
• Sunnyvale, CA: USD 232,000 - 258,000 per year.