Machine Learning Engineer • Staff

Who you are

You have a Master's degree in a relevant field such as Computer Science, Electrical Engineering, Mathematics, or Statistics, and at least 6 years of full-time software engineering experience focused on deep learning. Your proficiency in Python and PyTorch is complemented by your expertise in designing, debugging, and optimizing distributed deep learning systems. You have a strong ability to translate complex deep learning requirements into scalable solutions, ensuring that the systems you build are reliable and efficient.

You are familiar with distributed training frameworks such as DDP, DeepSpeed, FSDP, or TorchRec, and you have working experience with distributed training in PyTorch at scale, including data parallelism and model parallelism. Your background also includes familiarity with programming languages like C++, Go, or CUDA, which enhances your ability to optimize GPU/TPU training performance and data loading efficiency.

What you'll do

In this role, you will partner with stakeholders and lead team efforts to build and maintain machine learning backend services and solutions that support user-facing products and infrastructure tools across Uber. You will design and build tools that empower production teams to innovate and productionize state-of-the-art deep learning models. Your responsibilities will include developing and maintaining scalable, end-to-end deep learning training systems and frameworks, ensuring that distributed training tools are reliable and flexible for new production use cases.

Collaboration is key in this role, as you will work closely with cross-functional teams, including machine learning engineers, backend engineers, data scientists, and data engineers, to deliver robust ML solutions. You will be expected to ensure that the systems you develop can handle the demands of production environments while maintaining high performance and reliability.

What we offer

Uber provides a dynamic work environment where innovation is encouraged. You will have the opportunity to work on cutting-edge technologies and contribute to projects that have a significant impact on the company's operations. We value collaboration and support a culture of continuous learning and development, ensuring that you have the resources you need to succeed in your role. Additionally, you will be part of a team that is dedicated to pushing the boundaries of machine learning and its applications in real-world scenarios.