Senior AI Scientist, Generative ML for Code & IDE

Amazon Inc.; Seattle (WA); Hybrid; USD 167,000 - 226,000; Full time

Benefits offered by this job

Health insurance

RSUs

401(k) matching

Job summary

Amazon Web Services (AWS) is assembling an elite team for AI-driven development tools, including Kiro IDE and Amazon Q Developer. The Kiro LLM-Training team shapes next-gen code intelligence with cutting-edge RL and ML research, aiming to deliver production-ready AI tooling for developers at scale.

The role emphasizes reinforcing learning, large-scale systems, and collaboration with customers, with opportunities to publish at top conferences and impact millions of developers worldwide.

Qualifications

• PhD or Master’s degree with 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 Amazon scale.
• Deploy products that impact daily workflows of millions of developers.
• Advance AI and ML for intelligent code assistance.
• Publish and present work at premier ML/NLP conferences (NeurIPS, ICML, ICLR, ACL, EMNLP).
• Accelerate innovation by interfacing with customers to transition research to production.

Skills

Java

C++

Python

Education

PhD

Master's degree

Tools

R

scikit-learn

Spark MLLib

MxNet

TensorFlow

numpy

scipy

Job description

Amazon Web Services (AWS) is assembling an elite team for AI-driven development tools, including Kiro IDE and Amazon Q Developer. The Kiro LLM-Training team shapes next-gen code intelligence with cutting-edge RL and ML research, aiming to deliver production-ready AI tooling for developers at scale.

The role emphasizes reinforcing learning, large-scale systems, and collaboration with customers, with opportunities to publish at top conferences and impact millions of developers worldwide.