ML Model Serving Platform

Design a general-purpose ML model serving platform for non-LLM prediction serving used by many teams.

00

Practice checkpoints

The requirements are open as a taste. From the numbers onward, the full guide opens in the app.

  1. 01
    Clarify scope
  2. 02
    Requirements + scale
  3. 03
    API + data modelUnlocks in the app
  4. 04
    Draw architectureUnlocks in the app
  5. 05
    Deep diveUnlocks in the app
  6. 06
    Trade-off decisionUnlocks in the app
01

Requirements that shape the design

Do not only state requirements. Ask for them. Each card pairs the design constraint with a clarification question you can say out loud before drawing the architecture.

Functional requirements

01What kinds of models does this platform serve?

General-purpose prediction serving — ranking, vision, fraud/tabular — any framework (XGBoost, PyTorch, TF) behind one uniform predict API on a standard runtime; LLM token streaming is a different platform with different physics.

02How does a team ship a new model version?

Push the artifact to the registry with a version, a schema signature, and training metadata; deploying is promoting that registry entry — never rebuilding a service image around the weights.

03Do requests arrive with features, or does the platform fetch them?

Both: callers may send raw features, but the common case is entity ids — the platform multi-gets the rest from the feature store using the same feature definitions training used.

04What stands between a new version and live traffic?

A rollout ladder: shadow (mirror traffic, serve nothing), then a canary percentage judged on sliced metrics, then ramp — with rollback to the previous version possible at every rung in seconds.

05Hundreds of models from many teams — a dedicated fleet each?

No: hot models get dedicated replicas; the long tail is bin-packed many-to-a-GPU with per-model quotas, so one team’s spike cannot starve another team’s model.

06What does a caller get when inference cannot finish in time?

A degraded answer inside the deadline — a cached prediction or a declared default — never a hang; the caller’s timeout is a contract the platform designs around.

Out of scopeLLM and token-streaming inference (KV caches, per-token scheduling — the LLM Gateway question) · Training pipelines and offline feature computation (the Feature Store question owns point-in-time correctness) · Experiment analysis tooling beyond canary and A/B traffic splitting

Non-functional requirements

01What latency budget does the hottest caller hand us?

The hot caller’s 50 ms p99 is decomposed into owned budgets — feature fetch ~10 ms, batch wait capped at 5 ms, inference ~10 ms — each stage with its own monitor.

02A canary turns bad at 2 a.m. — how fast is it gone?

A bad version is out of traffic in seconds: rollback is a registry pointer swap to the previous version, which stays loaded and warm until the new one has soaked at 100%.

03The registry or the feature store degrades — do predictions stop?

Serving survives control-plane trouble: replicas keep answering with the already-loaded version, and feature-store misses fall back to defaults with the degradation logged.

04What GPU utilization justifies the bill?

Batching and bin-packing hold GPU utilization around 40–60% — lower means paying for idle silicon, near-saturation means no headroom when traffic spikes.

05How do we notice a model quietly rotting?

Every model logs sampled inputs and predictions by default, so drift against the training snapshot is detectable before any ground-truth label arrives.

Keep asking — the interview is a conversation

Real interviews probe far more than a tidy list. These are the scope questions that separate candidates who interrogate the problem from those who recite it.

  • Who are the callers and what are their latency SLAs — is there a 50 ms online path, or is everything near-real-time?
  • Which models actually earn a GPU, and which are fine on CPU?
  • Do requests carry full feature vectors, or entity ids that need a feature-store fetch on the hot path?
  • How many models and what traffic shape — a few hot ones, or hundreds of long-tail models?
  • When does ground truth arrive — clicks in minutes, chargebacks in weeks, or never?
01

Unlock the full playbook for ML Model Serving Platform

Numbers, architecture diagram, API and data model, deep dives, expected topics, self-check, whiteboard starter, and common mistakes unlock inside the app.

02

Numbers that force architecture decisions

Locked in the app

03

Architecture path

Locked in the app

04

API and data model

Locked in the app

05

Deep dive directions

Locked in the app

+Series

Practice the related series