Staff Software Engineer - AI Platform (Michelangelo)

Uber is seeking a Staff Software Engineer to join their AI Platform (Michelangelo) team, focusing on building and maintaining Machine Learning backend services and solutions. This role is critical in developing tools and frameworks that enable production teams across Uber to implement and deploy state-of-the-art deep learning models.

The position requires expertise in distributed deep learning systems, with a focus on PyTorch at scale. You'll be responsible for designing and maintaining scalable training systems that power Uber's machine learning infrastructure. The role involves close collaboration with various teams including ML engineers, data scientists, and backend engineers to deliver robust ML solutions.

The ideal candidate will have a Master's degree in Computer Science or related fields, along with 6+ years of software engineering experience specifically in deep learning. Strong proficiency in Python and PyTorch is essential, as is experience with distributed training frameworks and optimization of GPU/TPU training performance.

This is an excellent opportunity for someone passionate about large-scale machine learning systems, offering competitive compensation ($223,000-$248,000) plus equity and benefits. The role provides the chance to work on cutting-edge AI technology that impacts millions of users globally through Uber's platform.

Working in a hybrid environment across multiple locations (San Francisco, Sunnyvale, or Seattle), you'll be at the forefront of AI innovation, helping to scale and improve Uber's machine learning capabilities. The position offers significant technical challenges and the opportunity to work with some of the most advanced ML infrastructure in the industry.