Member of Technical Staff, ML Infrastructure

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

Job summary

Join Amazon's new research lab in San Francisco as a Member of Technical Staff, focused on AI infrastructure development. Be part of a talented team that shapes the future of AI by building cutting-edge tools for model training and evaluation. This role seeks individuals with deep software development experience and a passion for machine learning.

Qualifications

• 5+ years of professional software development experience required.
• Experience as a tech lead or engineering mentor is preferred.
• Ability to work with development processes like code reviews and testing.

Responsibilities

• Contribute to AI agent development by developing industry-leading tools.
• Develop tools for model training, evaluation, and key research workflows.

Skills

Proficiency in Python

Proficiency in Java

Proficiency in C++

Applied knowledge of machine learning

Experience with GPU training infrastructure

Expertise across the software development life cycle

Education

Bachelor's degree in Computer Science

Job description

Member of Technical Staff, ML Infrastructure

Job ID: 2999356 | Amazon.com Services LLC

Amazon has launched a new research lab in San Francisco to develop foundational capabilities for useful AI agents. We’re enabling practical AI to make our customers more productive, empowered, and fulfilled. In particular, our work combines large language models (LLMs) with reinforcement learning (RL) to solve reasoning, planning, and world modeling in both virtual and physical environments. Our research builds on that of Amazon’s broader AGI organization, which recently introduced Amazon Nova, a new generation of state-of-the-art foundation models (FMs).

Our lab is a small, talent-dense team with the resources and scale of Amazon. Each team in the lab has the autonomy to move fast and the long-term commitment to pursue high-risk, high-payoff research. We’re entering an exciting new era where agents can redefine what AI makes possible. We’d love for you to join our lab and build it from the ground up!

Key job responsibilities
You will contribute directly to AI agent development by developing industry-leading tools for model training, evaluation, and other key research workflows. You will be hired as a Member of Technical Staff.

BASIC QUALIFICATIONS

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

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 life cycle, including code reviews, build processes, testing, and operations

Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status.

Los Angeles County applicants: Job duties for this position include: work safely and cooperatively with other employees, supervisors, and staff; adhere to standards of excellence despite stressful conditions; communicate effectively and respectfully with employees, supervisors, and staff to ensure exceptional customer service; and follow all federal, state, and local laws and Company policies. Criminal history may have a direct, adverse, and negative relationship with some of the material job duties of this position. These include the duties and responsibilities listed above, as well as the abilities to adhere to company policies, exercise sound judgment, effectively manage stress and work safely and respectfully with others, exhibit trustworthiness and professionalism, and safeguard business operations and the Company’s reputation. Pursuant to the Los Angeles County Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records.

Pursuant to the San Francisco Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records.

Our inclusive culture empowers Amazonians to deliver the best results for our customers. If you have a disability and need a workplace accommodation or adjustment during the application and hiring process, including support for the interview or onboarding process, please visit https://amazon.jobs/content/en/how-we-hire/accommodations for more information. If the country/region you’re applying in isn’t listed, please contact your Recruiting Partner.

Posted: February 19, 2025 (Updated about 1 hour ago)

Posted: June 5, 2025 (Updated about 1 hour ago)

Posted: April 16, 2024 (Updated about 8 hours ago)

Posted: March 12, 2025 (Updated about 9 hours ago)

Posted: March 4, 2025 (Updated about 9 hours ago)

Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status.