Senior AI Scientist, LLMs for Code Intelligence (Hybrid)

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

Benefits offered by this job

Sign-on payments

RSUs

Health insurance

401(k) matching

Paid time off

Parental leave

Job summary

Amazon.com Services LLC is assembling a world-class team to build AI-driven development tools, including Kiro IDE and Amazon Q Developer, transforming how developers work across IDEs and web tools.

Join the AWS Kiro LLM-Training team to push RL/ML boundaries, ship scalable solutions, and present novel research at top conferences. Hybrid work near U.S. offices, with strong collaboration and mentorship opportunities.

Qualifications

• PhD or Master’s degree and 6+ years of applied research experience.
• Experience programming in Java, C++, Python or related language.

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 ML solutions at scale.
• Deploy products that impact developers’ workflows worldwide.
• Publish and present work at premier ML/NLP conferences (NeurIPS, ICML, ICLR, ACL, EMNLP).
• Accelerate innovation by transitioning research to production with customers.
• Collaborate across teams to advance AI tooling in IDE environments.

Skills

Java

C++

Python

Education

PhD or MS + 6+ yrs research

Tools

R

scikit-learn

Spark

TensorFlow

MxNet

Job description

Amazon.com Services LLC is assembling a world-class team to build AI-driven development tools, including Kiro IDE and Amazon Q Developer, transforming how developers work across IDEs and web tools.

Join the AWS Kiro LLM-Training team to push RL/ML boundaries, ship scalable solutions, and present novel research at top conferences. Hybrid work near U.S. offices, with strong collaboration and mentorship opportunities.