Member of Technical Staff - Applied Scientist (New PhD), AGI Autonomy

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

Job summary

An innovative research lab in San Francisco is seeking a Member of Technical Staff to drive AI agent development. This role offers the opportunity to work with cutting-edge technologies, combining large language models with reinforcement learning to create impactful AI solutions. As part of a small, dynamic team, you will engage in model training, dataset design, and optimization processes, contributing to groundbreaking research that aims to redefine AI capabilities. Join a forward-thinking organization committed to diversity and inclusion, where your contributions can shape the future of AI technology.

Qualifications

• PhD in Computer Science or a related field is required.
• Proficiency in programming languages such as Python, Java, or C++.

Responsibilities

• Contribute to AI agent development, including model training and dataset design.
• Engage in pre- and post-training optimization for AI models.

Skills

Python

Java

C++

Deep Learning Methods

Machine Learning

Education

PhD in Computer Science or related field

Tools

PyTorch

JAX

Job description

Member of Technical Staff - Applied Scientist (New PhD), AGI Autonomy

Job ID: 2912776 | 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 in an applied research role, including model training, dataset design, and pre- and post-training optimization. You will be hired as a Member of Technical Staff.

BASIC QUALIFICATIONS

• PhD degree in Computer Science or a related field
• Proficiency in Python, Java, C++, or related language
• Experience with deep learning methods and tools, e.g., PyTorch, JAX

PREFERRED QUALIFICATIONS

• Strong background in scientific research with a proven ability to generate and implement new ideas in machine learning
• Willingness to step outside typical role boundaries to get things done — every member of technical staff is expected to write code, design experiments, and interpret results
• Ability to communicate results and insights to both technical and non-technical audiences, including through presentations and written reports
• Capacity to work autonomously and with a small team to set and achieve a research agenda

Amazon is committed to a diverse and inclusive workplace. Amazon is an equal opportunity employer and does not discriminate on the basis of race, national origin, gender, gender identity, sexual orientation, protected veteran status, disability, age, or other legally protected status.