Member of Technical Staff - Reinforcement Learning, AGI Autonomy

Amazon; San Francisco (CA); On-site; USD 150,000 - 260,000; Full time

Job summary

A leading technology company seeks a Member of Technical Staff specializing in Reinforcement Learning. This role involves developing advanced LLMs, collaborating with cross-functional teams, and mentoring junior members. Candidates should possess a PhD or Master’s with extensive experience in machine learning, and proficiency in programming languages. The position offers a competitive salary and additional benefits.

Benefits

Equity

Sign-on bonuses

Qualifications

• 5+ years of applied research experience in machine learning.
• 3+ years building machine learning models for business use.
• Experience with neural networks and machine learning techniques.

Responsibilities

• Develop advanced multimodal Large Language Models (LLMs) for automation.
• Collaborate in agile teams to create automation agents.
• Mentor junior team members and contribute to team growth.

Skills

Machine Learning

Problem Solving

Communication

Education

PhD or Master’s in Computer Science, Machine Learning, or related field

Tools

Java

C++

Python

TensorFlow

PyTorch

Job description

Member of Technical Staff - Reinforcement Learning, AGI Autonomy

Join to apply for the Member of Technical Staff - Reinforcement Learning, AGI Autonomy role at Amazon.

The Amazon AGI SF Lab focuses on developing foundational capabilities for AI agents capable of acting in digital and physical environments. Our goal is to enable practical AI that enhances productivity and empowerment for our customers.

The lab is designed to foster rapid breakthroughs by combining startup agility with Amazon's resources. We maintain a lean team to maximize compute per individual, with autonomy to innovate and a long-term commitment to high-risk, high-reward research. Interested candidates can reach out via [e-mail].

Key job responsibilities

• Develop advanced multimodal Large Language Models (LLMs) to analyze workflows for automation.
• Collaborate in agile teams to create automation agents and launch them for millions of users.
• Work with cross-functional teams to solve complex GenAI problems.
• Design experiments to evaluate and improve algorithms and models.
• Plan long-term development of GenAI and identify new application opportunities.
• Communicate findings effectively to technical and non-technical audiences.
• Mentor junior team members and contribute to team growth.

Basic Qualifications

• PhD or Master’s with 5+ years of applied research experience.
• 3+ years building machine learning models for business use.
• Proficiency in Java, C++, Python, or similar languages.
• Experience with neural networks and machine learning techniques.

Preferred Qualifications

• PhD in Computer Science, Machine Learning, or related field, focused on Gen AI and reinforcement learning.
• Experience developing algorithms for supervised fine-tuning and reinforcement learning with human feedback.
• Strong Python skills and familiarity with frameworks like TensorFlow or PyTorch.
• Excellent problem-solving and creative thinking skills.
• Effective communication and teamwork abilities.
• Patents or publications in top-tier conferences/journals.

Amazon is an equal opportunity employer. We consider qualified applicants with criminal records, in accordance with local laws. For accommodations during the application process, visit https://amazon.jobs/content/en/how-we-hire/accommodations.

Salary range: $150,400 - $260,000 annually, depending on location, experience, and skills. Additional benefits may include equity and sign-on bonuses. For more details, visit https://www.aboutamazon.com/workplace/employee-benefits. The position remains open until filled. Apply through our career site.