Software Engineer - ML Michelangelo
Listed on 2026-02-18
-
Software Development
Machine Learning/ ML Engineer, AI Engineer, Data Engineer, Software Engineer
About the Role
Partners with stakeholders and leads team efforts to build and maintain Machine Learning backend services and solutions to support user-facing products, downstream services, or infrastructure tools and platforms used across Uber.
What the Candidate Will DoDesign and build tools to empower production teams to innovate and product ionize 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.
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 Py Torch
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.
Expertise in distributed training frameworks such as DDP, Deep Speed, FSDP, or Torch Rec.
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.)
For Sunnyvale, CA-based roles:
The base salary range for this role is USD $232,000 per year - USD $258,000 per year. 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. More details can be found at the following link
Uber is proud to be an Equal Opportunity/Affirmative Action 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-
#J-18808-Ljbffr(If this job is in fact in your jurisdiction, then you may be using a Proxy or VPN to access this site, and to progress further, you should change your connectivity to another mobile device or PC).