Senior Software Development Engineer , Stores Foundational AI - Rufus

Amazon.com, Inc.
Palo Alto, United States of America
16 days ago

Role details

Contract type
Internship / Graduate position
Employment type
Full-time (> 32 hours)
Working hours
Regular working hours
Languages
English
Experience level
Senior
Compensation
$ 262K

Job location

Palo Alto, United States of America

Tech stack

Artificial Intelligence
Code Review
Nvidia CUDA
Computer Programming
Software Design Patterns
Distributed Systems
Statistical Hypothesis Testing
Machine Learning
Performance Tuning
Software Engineering
Reinforcement Learning
Large Language Models
Information Technology
Build Process
Machine Learning Operations
TensorRT
Software Coding
Software Version Control
Programming Languages

Job description

We are building foundational LLMs for Amazon Stores that fuse world knowledge with deep e-commerce understanding to power next-generation shopping experiences. These systems continuously learn from real-world customer interactions to become more helpful, personalized, and context-aware over time.

We are looking for builders who are passionate about large-scale systems, AI innovation, and customer impact. You will work at the intersection of distributed systems, machine learning infrastructure, and science to bring frontier research-especially in post-training and reinforcement learning-into production at Amazon scale.

Key job responsibilities

  • Architect and build scalable ML infrastructure powering LLM training and post-training workflows, including supervised fine-tuning, reinforcement learning, and continuous learning from live traffic
  • Transform real-world customer interactions into high-quality training signals, enabling continuous model improvement and better customer experiences
  • Build and optimize post-training and RL systems, including reward modeling, policy optimization, data collection loops.
  • Drive experimentation and iteration velocity by building tooling and frameworks that enable rapid hypothesis testing, signal validation, and model quality improvements
  • Partner closely with applied scientists to translate frontier techniques (e.g., RLHF, agentic workflows, multi-turn optimization) into reliable, production-grade systems
  • Own systems end-to-end, including design, implementation, deployment, observability, and operational excellence
  • Raise the engineering bar through technical leadership, design reviews, and mentorship, influencing best practices across the organization

Requirements

  • 5+ years of non-internship professional software development experience
  • 5+ years of programming with at least one software programming language experience
  • 5+ years of leading design or architecture (design patterns, reliability and scaling) of new and existing systems experience
  • Experience as a mentor, tech lead or leading an engineering team
  • Experience with vLLM, SGLang, TensorRT or similar platforms in production environments
  • Experience with CUDA kernels or ML/low-level kernels, * 5+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience
  • Bachelor's degree in computer science or equivalent
  • Experience with Machine Learning and Large Language Model fundamentals, including architecture, training/inference lifecycles, and optimization of model execution

Benefits & conditions

3.53.5 out of 5 stars Palo Alto, CA $193,300 - $261,500 a year - Full-time, Pulled from the full job description

  • AD&D insurance
  • Parental leave
  • Health insurance
  • 401(k) matching
  • Paid time off
  • Vision insurance
  • Dental insurance, 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, Palo Alto - 193,300.00 - 261,500.00 USD annually USA, WA, Seattle - 168,100.00 - 227,400.00 USD annually

Apply for this position