AI Infra Interviews logo

AI Infrastructure System Design

34 questions
0 of 34 done · 10 unlocked for you
DONEUNLOCKEDLOCKED

AI 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.

  1. Inference Platform Architecture

    Assign responsibilities and failure handling to serving components.

    Public concept
  2. 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
  3. 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
  4. 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
  5. 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
  6. 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
  7. 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
You have 10 free answers unlocked here.Sign in free for 10 more · 14 are premium.
01–14Foundationsbuild the vocabulary and basic mechanisms0/14 done
15–26Core loopsapply the ideas to worked interview questions0/12 done
27–34Field scenariosreason through constraints and operational failure scenarios0/8 done
PRACTICE TESTStudied this track? Prove it: a timed AI Infrastructure System Design test.rotating questions · every answer explained · free with an accountStart →

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.

Foundational
Inference Platform ArchitectureAn inference platform connects API requests to model replicas. Learn how routing, batching, memory limits and scaling work together to meet a latency target.
Advanced
🔒 Premium
Request Routing and Load Balancing for LLMsAn LLM router chooses a compatible replica, weighing queue time against reusable prompt state. Cached work helps only when the extra wait does not erase the saving.
Core
Sign in
GPU Job Scheduler DesignA GPU scheduler decides which jobs run, where they run and when resources can be reclaimed. Its placement rules must account for whole jobs, topology and tenant policy.
Advanced
🔒 Premium
Training Cluster Design at 10k GPUsSize a training cluster from the model, data and completion deadline. Then check the network, storage, power and recovery capacity needed to keep those GPUs productive.
Advanced
🔒 Premium
Serverless GPU PlatformsA serverless GPU platform hides allocation and startup from its users. The design balances cold-start latency, tenant isolation and the cost of idle capacity.
Advanced
🔒 Premium
Multi-Tenant Fine-Tuning ServiceA fine-tuning service turns a tenant’s dataset into an evaluated model or adapter. Its pipeline must protect data while scheduling training and serving the result efficiently.
Advanced
🔒 Premium
Evaluation and Data Pipeline InfrastructureData pipelines prepare traceable training inputs; evaluation pipelines compare model behavior. Both need versioned inputs, reproducible execution and useful failure reporting.
Advanced
🔒 Premium
Capacity and BackpressureBackpressure limits accepted work when a service cannot keep up. Bounded queues and clear retry rules prevent overload from spreading through the request path.
Unlock all 34 answers · ₹2,000 / $25