Job description

AWS Agentic AI is looking for a Principal Applied Scientist who will partner with world-class scientists and engineers to pioneer the next generation of AI-driven development tools. Join the Amazon Kiro LLM-Training team and help create the most efficient coding agents for Kiro that are transforming the software development landscape. Key job responsibilities As a key member of our team, you'll be at the forefront of innovation in agentic RL space: - Push the boundaries of reinforcement learning and post-training methodologies for large language models specialized in code intelligence - Invent and implement state-of-the-art agentic RL training recipes that operate at unprecedented Amazon scale - Deploy revolutionary products that directly impact the daily workflows of millions of developers worldwide - Break new ground in AI and machine learning, challenging what's possible in intelligent code assistance - Publish and present your pioneering work at premier ML conferences (NeurIPS, ICML, ICLR ) - Accelerate innovation by working directly with customers to rapidly transition research breakthroughs into production systems About the team Kiro team is reimagining the builder experience through generative AI and foundation models. We're leveraging the latest advances in AI to transform how engineers work from IDE environments to web-based tools and services, empowering developers to tackle projects of any scale with unprecedented efficiency. - PhD, or Master's degree and 10+ years of CS, CE, ML or related field experience - 5+ years of hands-on large scale training experience - Experience in patents or publications at top-tier peer-reviewed conferences or journals - Experience programming in C++, Python or related language - Prior experience at training or inference large scale models Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status. Los Angeles County applicants: Job duties for this position include: work safely and cooperatively with other employees, supervisors, and staff; adhere to standards of excellence despite stressful conditions; communicate effectively and respectfully with employees, supervisors, and staff to ensure exceptional customer service; and follow all federal, state, and local laws and Company policies. Criminal history may have a direct, adverse, and negative relationship with some of the material job duties of this position. These include the duties and responsibilities listed above, as well as the abilities to adhere to company policies, exercise sound judgment, effectively manage stress and work safely and respectfully with others, exhibit trustworthiness and professionalism, and safeguard business operations and the Company’s reputation. Pursuant to the Los Angeles County Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records. Our inclusive culture empowers Amazonians to deliver the best results for our customers. If you have a disability and need a workplace accommodation or adjustment during the application and hiring process, including support for the interview or onboarding process, please visit amazon.jobs for more information. If the country/region you’re applying in isn’t listed, please contact your Recruiting Partner. 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 amazon.jobs . USA, CA, Santa Clara - 228,700.00 - 309,400.00 USD annually USA, NY, New York - 218,800.00 - 295,900.00 USD annually USA, WA, Seattle - 198,900.00 - 269,000.00 USD annually

Requirements

Must-haves

Key job responsibilities
As a key member of our team, you'll be at the forefront of innovation in agentic RL space:
Push the boundaries of reinforcement learning and post-training methodologies for large language models specialized in code intelligence
Invent and implement state-of-the-art agentic RL training recipes that operate at unprecedented Amazon scale
Deploy revolutionary products that directly impact the daily workflows of millions of developers worldwide
Break new ground in AI and machine learning, challenging what's possible in intelligent code assistance
Publish and present your pioneering work at premier ML conferences (NeurIPS, ICML, ICLR )
Accelerate innovation by working directly with customers to rapidly transition research breakthroughs into production systems
About the team
PhD, or Master's degree and 10+ years of CS, CE, ML or related field experience
5+ years of hands-on large scale training experience
Experience in patents or publications at top-tier peer-reviewed conferences or journals
Experience programming in C++, Python or related language
Prior experience at training or inference large scale models
Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status.
USA, CA, Santa Clara - 228,700.00 - 309,400.00 USD annually
USA, NY, New York - 218,800.00 - 295,900.00 USD annually
USA, WA, Seattle - 198,900.00 - 269,000.00 USD annually

Nice-to-haves

None flagged.

Application prep

Use the role’s language and application details while tailoring your materials.

Keywords to mirror

awsaillmmachine learningmlpython

Platform: amazon
Effort: medium

Benefits

Benefits detected in the listing.

Health Insurance

Dental

Vision

401(K)

Paid Time Off

Parental Leave