Senior Applied Scientist, LLM Code Agents, Kiro Science

Amazon; New York (NY); Hybrid; USD 184,000 - 249,000; Full time

Benefits offered by this job

Health insurance

401(k) matching

Paid time off

Parental leave

Job summary

Amazon is seeking an experienced ML/AI researcher to join the Kiro LLM-Training team at the AWS Development Center. You will push the boundaries of reinforcement learning and post-training techniques for code-focused language models, and deploy advanced ML solutions at scale.

The role involves collaborating with customers, presenting at top conferences, and helping bring research breakthroughs into production systems while enjoying a flexible, hybrid work model near AWS offices in New York.

Qualifications

• 3+ years of building machine learning models for business applications.
• PhD or Master’s degree with 6+ years of applied research experience.
• Experience programming in Java, C++, Python or related language.
• Experience with neural deep learning methods and machine learning.

Responsibilities

• Push the boundaries of RL and post-training methods for large language models focused on code intelligence.
• Invent and implement state-of-the-art ML solutions at scale for AWS.
• Deploy products that impact daily workflows of millions of developers.
• Publish and present work at premier ML/NLP conferences (NeurIPS, ICML, ICLR, ACL, EMNLP).
• Work with customers to transition research breakthroughs into production systems.

Skills

Java

C++

Python

Deep learning

Education

PhD

Master's degree

Tools

TensorFlow

PyTorch

JAX

Job description

Job ID: 10526425 | Amazon Development Center U.S., Inc.

Amazon Web Services (AWS) is assembling an elite team of 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 groundbreaking generative AI technologies including Kiro IDE and Amazon Q Developer that are transforming the software development landscape.

Key job responsibilities

• 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 machine learning solutions 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 and NLP conferences (NeurIPS, ICML, ICLR , ACL, EMNLP)
• Accelerate innovation by working directly with customers to rapidly transition research breakthroughs into production systems

About the team

The AWS Developer Agents and Experiences (DAE) 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.

Broadly, AWS Utility Computing (UC) provides product innovations — from foundational services such as Amazon’s Simple Storage Service (S3) and Amazon Elastic Compute Cloud (EC2), to consistently released new product innovations that continue to set AWS’s services and features apart in the industry. As a member of the UC organization, you’ll support the development and management of Compute, Database, Storage, Internet of Things (Iot), Platform, and Productivity Apps services in AWS. Within AWS UC, Amazon Dedicated Cloud (ADC) roles engage with AWS customers who require specialized security solutions for their cloud services.

Why AWS?

Amazon Web Services (AWS) is the world’s most comprehensive and broadly adopted cloud platform. We pioneered cloud computing and never stopped innovating — that’s why customers from the most successful startups to Global 500 companies trust our robust suite of products and services to power their businesses.

Inclusive Team Culture

Here at AWS, it’s in our nature to learn and be curious. Our employee‑led affinity groups foster a culture of inclusion that empower us to be proud of our differences. Ongoing events and learning experiences, including our Conversations on Race and Ethnicity (CORE) and AmazeCon conferences, inspire us to never stop embracing our uniqueness.

Mentorship & Career Growth

We’re continuously raising our performance bar as we strive to become Earth’s Best Employer. That’s why you’ll find endless knowledge‑sharing, mentorship and other career‑advancing resources here to help you develop into a better‑rounded professional.

Work/Life Balance

We value work‑life harmony. Achieving success at work should never come at the expense of sacrifices at home, which is why we strive for flexibility as part of our working culture. When we feel supported in the workplace and at home, there’s nothing we can’t achieve in the cloud.

Hybrid Work

We value innovation and recognize this sometimes requires uninterrupted time to focus on a build. We also value in‑person collaboration and time spent face‑to‑face. Our team affords employees options to work in the office every day or in a flexible, hybrid work model near one of our U.S. Amazon offices.

Basic Qualifications

• - 3+ years of building machine learning models for business application experience
• - PhD, or Master’s degree and 6+ years of applied research experience
• - Experience programming in Java, C++, Python or related language
• - Experience with neural deep learning methods and machine learning

Preferred Qualifications

• - Have peer‑reviewed scientific contributions in premier journals and conferences

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 https://amazon.jobs/content/en/how-we-hire/accommodations for more information. If the country/region you’re applying in isn’t listed, please contact your Recruiting Partner.

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 .

Job details

• USA, WA, Bellevue
• USA, NY, New York

Salary ranges

• USA, CA, Santa Clara - 192,200.00 - 260,000.00 USD annually
• USA, NY, New York - 183,800.00 - 248,700.00 USD annually
• USA, WA, Bellevue - 167,100.00 - 226,100.00 USD annually

Share this job

Important FAQs for current Government employees

Before proceeding, please review the following FAQs

https://www.amazon.jobs/en/faqs#faqs-for-us-government-employees

Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status.