Amazon is looking for a passionate and inventive scientist to advance the science in foundational models and Agentic AI. Specifically, as part of our science team in Amazon AWS Agentic AI, you will lead the research and development of techniques for efficient and effective Agent optimization/learning, through various techniques ranging from model fine-tuning (e.g. RL) to context optimization, as a foundational layer for AWS customers to build reliable and performant Agents.
You will have the opportunity to take a product from zero to one by influencing directly the product and science roadmap, and impact millions of our customers. You will also gain hands on experience with Amazon's large-scale computing resources to accelerate advances in foundation models.
Key job responsibilities
* Develop short-term and long-term science roadmap for research in the broad area of Agent optimization/learning, which could range from RL reward shaping, training efficiency optimization, to automatic prompt optimization and techniques to learn from agent memory.
* Develop concrete science plan, implement and validate research idea before moving it to production
* Collaborate across product and engineering teams to transfer science innovations into AWS customer facing product.
BASIC QUALIFICATIONS
- PhD, or Master's degree and 4+ years of CS, CE, ML or related field experience
- Experience in patents or publications at top-tier peer-reviewed conferences or journals
- Experience programming in Java, C++, Python or related language
PREFERRED QUALIFICATIONS
- Experience with techniques like RLVR, reward modeling, inference optimization, agent task/environment synthesis.
The base salary range for this position is listed below. Your Amazon package will include sign-on payments and restricted stock units (RSUs). Final compensation will be determined based on factors including experience, qualifications, and location. Amazon also offers comprehensive benefits including health insurance (medical, dental, vision, prescription, Basic Life & AD&D insurance and option for Supplemental life plans, EAP, Mental Health Support, Medical Advice Line, Flexible Spending Accounts, Adoption and Surrogacy Reimbursement coverage), 401(k) matching, paid time off, and parental leave. Learn more about our benefits at https://amazon.jobs/en/benefits.
USA, CA, Santa Clara - 171,600.00 - 222,200.00 USD annually