Member of Technical Staff, Multimodal Reasoning - Applied Science , AGI Autonomy

Amazon • San Francisco, CA

2d • $255,000 - $345,000

About The Position

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. Our work leverages large
vision language models (VLMs) with reinforcement learning (RL) and world
modeling to solve perception, reasoning, and planning to build useful enterprise
agents.

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.

Requirements

• 5+ years' experience building machine learning models
• PhD or Master's degree in computer science or related field
• Proficiency in Python, Java, C++, or related language
• Experience with deep learning methods and tools, e.g., PyTorch, JAX

Nice To Haves

• Background in scientific research with a proven ability to generate and implement new ideas in machine learning
• Experience with post-training of large Vision Language Models (VLMs).
• 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
• Ability to think big about the arc of development of AI over a multi-year horizon, and identify new opportunities to apply these technologies to solve real-world problems
• Capacity to mentor and guide junior scientists and engineers, and contribute to the overall growth and development of the team

Responsibilities

• You will lead our efforts to improve the multi-model perception and reasoning abilities of our AI agent in an applied research role.
• Responsibilities including model training, dataset design, and pre- and post-training optimization.
• You will be hired as a Member of Technical Staff.

Benefits

• Our compensation reflects the cost of labor across several US geographic markets.
• The base pay for this position ranges from $255,000/year in our lowest geographic market up to $345,000/year in our highest geographic market.
• Pay is based on a number of factors including market location and may vary depending on job-related knowledge, skills, and experience.
• Amazon is a total compensation company.
• Dependent on the position offered, equity, sign-on payments, and other forms of compensation may be provided as part of a total compensation package, in addition to a full range of medical, financial, and/or other benefits.
• For more information, please visit https://www.aboutamazon.com/workplace/employee-benefits.

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level