Member of Technical Staff, ML Infrastructure

Amazon; San Francisco (CA); On-site; USD 130,000 - 180,000; Full time

Job summary

Amazon is seeking a Member of Technical Staff for ML Infrastructure to develop AI agents that enhance productivity. The role requires significant software development experience and proficiency in programming languages. Join a talented team working on cutting-edge AI research with autonomy and Amazon's resources.

Qualifications

• 5+ years of professional software development experience.
• Applied knowledge of machine learning theory and practice.
• Experience as a tech lead or engineering mentor.

Responsibilities

• Develop AI agents using large language models and reinforcement learning.
• Optimize GPU training infrastructure.
• Engage in the software development lifecycle including code reviews and testing.

Skills

Software Development

Python

Java

C++

Education

Bachelor's degree in Computer Science or related field

Tools

GPU training infrastructure

Job description

Member of Technical Staff, ML Infrastructure

Job ID: 2981841 | Amazon.com Services LLC

We are developing useful AI agents to empower customers and enhance productivity. Our work combines large language models (LLMs) with reinforcement learning (RL) to address reasoning, planning, and world modeling in virtual and physical environments. Our research builds on Amazon’s broader AGI initiatives, including Amazon Nova, a new generation of foundation models (FMs).

Our lab is a small, talented team with Amazon's resources, operating with autonomy to pursue high-risk, high-reward research. We are entering an era where AI agents can significantly expand what is possible. Join us to help build this innovative future!

BASIC QUALIFICATIONS

• 5+ years of professional software development experience
• Proficiency in Python, Java, C++, or related languages

PREFERRED QUALIFICATIONS

• Bachelor's degree in Computer Science or related field
• Applied knowledge of machine learning theory and practice
• Experience with GPU training infrastructure and optimization
• Experience as a tech lead or engineering mentor
• Expertise across the software development lifecycle, including code reviews, build processes, testing, and operations

Amazon is an equal opportunity employer and values diversity. We consider qualified applicants with arrest and conviction records in accordance with applicable laws. For accommodations during the hiring process, please visit this link.