Senior AI Systems Engineer, LLM Infrastructure

Amazon

Palo Alto (CA)

On-site

USD 193,000 - 262,000

Full time

14 days+
Application generator

Get a reply from this employer — a resume and cover letter tailored to exactly what they’re hiring for.

Get past ATS filters

Job summary

Amazon is seeking a Senior Software Development Engineer for Stores Foundational AI - Rufus in Palo Alto, CA. You will architect scalable ML infrastructure for LLM training, post-training workflows, and live-traffic data signals to improve customer experiences.

The role demands 5+ years in software development, strong systems design and leadership, and hands-on experience with ML platforms like vLLM, SGLang, TensorRT, and CUDA kernels.

Qualifications

  • 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.
  • Experience with Machine Learning and Large Language Model fundamentals, including architecture, training/inference lifecycles, and optimization of model execution

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

Education

Bachelor's degree in computer science or equivalent

Tools

vLLM
SGLang
TensorRT
CUDA

Job description

Amazon is seeking a Senior Software Development Engineer for Stores Foundational AI - Rufus in Palo Alto, CA. You will architect scalable ML infrastructure for LLM training, post-training workflows, and live-traffic data signals to improve customer experiences.

The role demands 5+ years in software development, strong systems design and leadership, and hands-on experience with ML platforms like vLLM, SGLang, TensorRT, and CUDA kernels.

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

Software Engineer II — AI & LLM Platforms
Software Engineer II — AI & LLM Platforms

Amazon • Palo Alto (CA)

On-site
USD 165,000 - 224,000
Health insurance
401(k) matching
Paid time off
+1
Lead Applied AI Scientist – LLM/VLM for Shopping
Lead Applied AI Scientist – LLM/VLM for Shopping

Amazon • San Francisco (CA), Northern (KY)

Hybrid
USD 211,000 - 286,000
Senior AI Leader, Shopping Experiences (LLM/VLM)
Senior AI Leader, Shopping Experiences (LLM/VLM)

Amazon • Arlington (VA), Northern (KY)

Hybrid
USD 219,000 - 340,000
Health insurance
401(k) matching
Paid time off
+1
Staff ML Systems Engineer - LLM Serving & RL
Staff ML Systems Engineer - LLM Serving & RL

Prime Intellect • San Francisco (CA), Northern (KY)

Hybrid
USD 150,000 - 300,000
Remote option
Visa sponsorship
Relocation support
+2
Senior ML Engineer: Frontier LLMs & RL Gym Dev (Onsite)
Senior ML Engineer: Frontier LLMs & RL Gym Dev (Onsite)

Amazon.com Services LLC • Sunnyvale (CA)

On-site
USD 193,000 - 262,000
Sign-on payments
Restricted stock units (RSUs)
Health insurance
+7
Senior Vector Search Engineer – ML/LLM & Scalable Infra
Senior Vector Search Engineer – ML/LLM & Scalable Infra

Amazon • Palo Alto (CA)

On-site
USD 193,300 - 261,500
Health insurance
401(k) matching
Paid time off
+2
Staff Engineer, Generative AI & LLM Systems
Staff Engineer, Generative AI & LLM Systems

Amazon • Sunnyvale (CA)

On-site
USD 180,000 - 240,000
Senior AI Platform Engineer - LLMs & MLOps
Senior AI Platform Engineer - LLMs & MLOps

Namely • Mountain View (CA)

On-site
USD 120,000 - 160,000
Senior Applied Scientist, Next-Gen LLMs
Senior Applied Scientist, Next-Gen LLMs

Amazon • Sunnyvale (CA)

On-site
USD 192,000 - 260,000
Health insurance
401(k) matching
Paid time off
AI Services SDE: Build Scalable ML/LLM Systems
AI Services SDE: Build Scalable ML/LLM Systems

Amazon Web Services (AWS) • Seattle (WA)

On-site
USD 144,000 - 194,000