Uber · 16 hours ago
Staff Software Engineer - ML MichaelAngelo Team
Sunnyvale, CA
Full-time
Onsite
Senior Level, Lead/Staff
$232K/yr - $258K/yr
6+ years expUber is a leading technology company focused on building innovative solutions for transportation and logistics. They are seeking a Staff Software Engineer to lead the development of Machine Learning backend services and solutions that support user-facing products and infrastructure across the company.
TransportationSoftwareAppsLogisticsMobile AppsRide Sharing
Responsibilities
Design and build tools to empower production teams to innovate and productionize state-of-the-art deep learning models at Uber
Develop and maintain scalable, end-to-end deep learning training systems and frameworks
Ensure distributed training tools are reliable, efficient, flexible to use for new production use cases
Collaborate with cross-functional teams including machine learning engineers, backend engineers, data scientists, and data engineers to deliver robust ML solutions for Uber
Qualification
Required
Master in relevant fields (CS, EE, Math, Stats, etc.) AND 6-years full-time Software Engineering work experience in deep learning
Proficiency in Python and PyTorch
Expertise in designing, debugging, and optimizing distributed deep learning systems
Working experience of distributed training in PyTorch at Scale (e.g., data parallelism, model parallelism)
Strong ability to translate complex DL requirements and problems into scalable solutions
Preferred
Expertise in distributed training frameworks such as DDP, DeepSpeed, FSDP, or TorchRec
Familiarity with C++, Go or CUDA programming
Expertise in optimizing GPU/TPU training performance and data loading efficiency
Familiarity with large-scale distributed infrastructure tools like Ray, OpenAI Triton, PyTorch Lightning
Built and deployed end-to-end machine learning systems in production
Experience training large models (10B+ parameters), such as large recommendation systems or large language models (LLMs)
PhD in relevant fields (CS, EE, Math, Stats, etc.)
Benefits
Participate in Uber's bonus program
May be offered an equity award & other types of comp
Eligible to participate in a 401(k) plan
Eligible for various benefits
Company
Uber
Uber develops, markets, and operates a ride-sharing mobile application that allows consumers to submit a trip request.
10001+ employees
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 CompanyTotal Funding
$35.56BKey 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
Dara Khosrowshahi
CEO
Prashanth Mahendra -Rajah
Chief Financial Officer
Recent News
2026-02-12
2026-02-12
Company data provided by crunchbase