Profile cover
S

AI Speculative Decoding Implementation Engineer

Stella Tran

Full-time · Job · Los_Angeles

About the role

Speculative decoding is one of the most impactful LLM serving optimizations of the past two years: use a small draft model to generate candidate tokens, verify them in parallel with the large model, and achieve 2-4x throughput improvement without any accuracy loss. The papers make it look straightforward. The production implementation is not. Our serving infrastructure handles 140 million LLM API calls per month across 12 different model families. We want to implement speculative decoding across our entire serving fleet — but the implementation details are genuinely hard: draft model selection per request type, acceptance criteria tuning, handling of streaming responses where rejected tokens affect perceived latency, and the monitoring needed to detect when the draft model is hurting rather than helping. This is a senior IC role requiring both deep ML systems knowledge and production serving experience. You'll implement the speculative decoding system, design the experiments that tune its parameters, and build the monitoring that tracks its impact across request types. The target: 2.5x throughput improvement at constant quality on our top-5 served models.

Responsibilities

  • Implement speculative decoding across production LLM serving infrastructure
  • Design draft model selection strategy per request type and model family
  • Build acceptance criteria tuning pipeline with continuous quality validation
  • Handle streaming response edge cases where token rejection affects perceived latency
  • Create monitoring tracking throughput gains, quality impact, and draft acceptance rate
  • Run A/B experiments validating speedup vs. quality trade-offs across model families

Requirements

  • 6+ years in ML systems or LLM inference engineering
  • Deep understanding of transformer inference, KV caching, and batching strategies
  • Experience with speculative decoding or other inference acceleration techniques
  • Strong Python and PyTorch skills with GPU optimization knowledge
  • Track record implementing low-level serving optimizations in production

Benefits

  • Fully remote with flexible schedule
  • Top-of-market compensation
  • Comprehensive health benefits
  • $5,000 annual equipment and learning budget
  • Equity participation

Keep exploring related roles

Use these topic pages to discover more roles in the same hiring cluster.

Job Type

Full-time

Level

Senior

Language

English

Salary Range

$160k – $200k / year

AI Expertise

AI & Machine Learning Engineers MLOps & AI Infrastructure

Skills

python pytorch docker aws kubernetes fastapi

Ready to apply for this role?

Create a free talent account in under 2 minutes.

  • Apply to verified AI companies
  • Get AI-matched job recommendations
  • Message hiring managers directly
  • Build your public AI talent profile

Interested in this role? Create your free profile to apply.

Apply — create free profile