Member of Technical Staff - Reinforcement Learning (Simulation), AGI Autonomy (AI)

TL;DR

Member of Technical Staff (AI, Reinforcement Learning): Designing and implementing highly scalable and realistic simulation environments for training AI agents using reinforcement learning, with an accent on generative AI approaches for automated environment creation and performance bottleneck resolution. Focus on solving challenging problems at the intersection of generative AI and reinforcement learning to build foundational capabilities for useful AI agents.

Location: Based in San Francisco, CA. Compensation varies by U.S. geographic market.

Salary: $120,000–$350,000/year

Company

Amazon has launched a new research lab in San Francisco to develop foundational capabilities for useful AI agents, combining large language models (LLMs) with reinforcement learning.

What you will do

• Design and implement highly scalable and realistic simulation environments for training agents using reinforcement learning.
• Develop and deploy generative AI approaches to automate the creation of diverse and complex simulation environments.
• Analyze, troubleshoot, and profile complex machine learning systems within the simulation context, identifying and resolving performance bottlenecks.
• Collaborate with cross-functional teams to identify and solve challenging problems at the intersection of generative AI and reinforcement learning.

Requirements

• PhD, Master's degree, or 5+ years of applied research experience.
• Experience programming in C++, Python or related language.
• Experience with simulation and synthetic data generation.
• Experience with neural deep learning methods and machine learning.

Nice to have

• PhD in Computer Science, Machine Learning, or a related field.
• 3+ years' experience building machine learning models (includes internships).
• Demonstrated experience in developing and implementing simulation environments or synthetic data generation for reinforcement learning.
• Strong programming skills in Python and experience with deep learning frameworks such as TensorFlow or PyTorch.
• Excellent problem-solving, communication, and collaboration skills.

Culture & Benefits

• Small, talent-dense team with autonomy and long-term commitment to high-risk, high-payoff research.
• Opportunity to build foundational capabilities for useful AI agents from the ground up.
• Inclusive culture empowering Amazonians to deliver best results.
• Full range of medical, financial, and other benefits.