Member of Technical Staff - Inference Research

Modal Modal · Data AI · New York, NY · Engineering

Research role focused on LLM inference optimization, including speculative decoding, quantization, KV-cache management, and autoscaling. The role involves training custom speculators, collaborating with customers, and turning frontier serving techniques into products. Requires a strong background in LLM serving stack and a track record of shipping research or systems.

What you'd actually do

  1. Own end-to-end inference research bets: speculative decoding, disaggregated prefill/decode, quantization (FP8, INT4), KV-cache and memory management, autoscaling for spiky serverless traffic, and whatever else the research agenda calls for.
  2. Train custom speculators against production traffic and feed what you learn back into target models -- acceptance length is the metric that decides the win.
  3. Work directly with customers alongside our Forward Deployed Engineers to deploy and tune models, and bring what you learn back into the research.
  4. Carry and expand collaborations with outside research labs, for example: our work with ZLab on DFlash, a speculator design built on KV injection and blockwise parallel drafting; our work with SGLang on specdec and multimodal inference performance; our work on Flash Attention 4 kernels.
  5. Work with engineering to turn frontier serving techniques into products: primitives for disaggregation, fast weight refresh for models that keep training after deployment, observability for quality and latency in production, or even a next-generation inference engine.

Skills

Required

  • LLM inference research
  • LLM serving stack
  • kernels
  • quantization
  • schedulers
  • autoscaling
  • shipping research or systems

Nice to have

  • speculative decoding
  • disaggregated prefill/decode
  • KV-cache and memory management
  • FP8
  • INT4
  • Flash Attention 4 kernels
  • multimodal inference

What the JD emphasized

  • inference research bets
  • speculative decoding
  • disaggregated prefill/decode
  • quantization
  • KV-cache and memory management
  • autoscaling
  • production traffic
  • deploy and tune models
  • frontier serving techniques
  • LLM inference
  • LLM serving stack
  • shipping research or systems

Other signals

  • LLM inference research
  • cost per token
  • tail latency
  • shipping research or systems