AI Infrastructure System Design
34 questionsAI Infrastructure System Design Interview Questions
Design an inference platform at 10k requests per second, a 10,000-GPU training cluster, a job scheduler with preemption and checkpointing, a serverless GPU runtime with sub-second cold starts, a multi-tenant fine-tuning service, an eval pipeline. The whiteboard round at OpenAI, Anthropic, Baseten and Together.
Learn how AI infrastructure works through explanations, worked examples and diagrams, then practise applying it to interview questions.
A starting path through this topic
Start with a mechanism, then use a worked problem to explain a decision. Access labels below describe what is available without an account.
- Inference Platform Architecture
Assign responsibilities and failure handling to serving components.
Public concept - Walk me through an inference platform for a hosted LLM. What are the pieces, and what does each one do?
Follow a hosted request through routing, scheduling and model execution.
Public answer - You have 45 minutes and the prompt is 'design our serving platform'. How do you run the round?
Practise allocating a design discussion around requirements and decisions.
Public answer - Design a job scheduler for 100,000 jobs on a shared GPU cluster, with preemption and checkpointing. Show me the state machine.
Show scheduling, preemption and checkpoint recovery as state transitions.
Public answer - Design a serverless GPU platform: a customer deploys a function with a model and pays per second. Where does the cold start budget go?
Account for model loading and idle capacity in a serverless design.
Public answer - Design the eval pipeline for a frontier model: thousands of evals per checkpoint, sharded inference, caching, reproducible results.
Preserve reproducibility while distributing evaluation work.
Public answer - Design the pipeline that produces 15 trillion training tokens: ingest, dedup, tokenize, shard, serve. Throughput per stage.
Budget ingestion, deduplication, tokenization and data delivery separately.
Public answer
The concepts behind AI Infrastructure System Design
Review the ideas used in these questions. Each card shows whether the explanation is free, requires sign-in or is part of Premium.
