AI Infra Interviews logo

Baseten Napkin Math, Cost & Capacity interview questions

Practise napkin math, cost & capacity for Baseten AI Infrastructure Engineer roles. Memory footprints, 6ND, arithmetic intensity and the ridge point, bandwidth-bound decode, communication volume, GPU counts and time to train, cost per million tokens, TCO and buy versus rent. The estimation round almost every AI infra loop includes, with every assumption stated. Questions tagged to Baseten appear first, followed by additional practice from our Napkin Math, Cost & Capacity track. Company tags indicate preparation relevance, not confirmed interview appearances. Confirm the current interview format with your recruiter.

PREPARATION CONTEXT · Inference stack from developer experience to Kubernetes orchestration and routing. Sources and confidence in the Baseten guide →

Napkin Math, Cost & Capacity questions tagged to Baseten

6 questions · 3 unlocked for you

Additional Napkin Math, Cost & Capacity practice

These questions cover the same topic and are not tagged to Baseten. Use them to practise applying the ideas in a different situation.

15 questions · 7 unlocked for you

Concepts for Napkin Math, Cost & Capacity preparation

These concept explanations cover the ideas used in the questions. Check each card for its access level.

Foundational
Model Memory FootprintSeparate weight payload, growing request state and training allocations. Work through exact units and per-rank budgets, then check the live peak.
Foundational
KV Cache SizingDerive cache payload from the stored attention state, then account for mixed sequence lengths, block allocation and per-device limits. Worked examples distinguish exact bytes from deployable capacity.
Foundational
Training FLOPs: 6NDDerive the six from one forward and two backward matrix products. Count training tokens, attention and recomputation consistently before turning model work into a schedule.
Core
Sign in
GPU-Hours and Time to TrainConvert a stated work budget into elapsed time and GPU-hours. Vary utilization, invert for a deadline, and separate productive time from calendar delays without counting them twice.
Core
Sign in
Arithmetic Intensity by OperationCount the FLOPs and memory transfers of a projection, attention and a residual add. Learn where the batch-size shortcut works, where it breaks, and which hardware roof applies.
Advanced
🔒 Premium
Bandwidth-Bound Decode ThroughputEstimate decode throughput from bytes per step, then separate aggregate output from per-user pace. A finite-memory example shows how batching, cache reads and quantization change the bound.
Advanced
🔒 Premium
Communication Volume EstimatesEstimate communication by naming the tensor, collective, group and synchronization schedule. Work through per-rank ring bytes, conditional sharding costs, activation exchanges and routing imbalance before converting traffic into exposed time.
Advanced
🔒 Premium
Cost per Million TokensCost per million tokens connects fleet spending to useful output. Keep the time window, token definition and utilization assumptions consistent when comparing deployments.
BASETEN NAPKIN MATH, COST & CAPACITY FAQ

This collection includes practice questions such as "How big is the KV cache for Llama 3.1 70B at a 128k context?", "How many tokens per second can a 70B model generate for a single user on H100s?", "What does it cost per million output tokens to serve a 70B model on eight H100s?". Company tags describe relevance to preparation; they do not confirm that Baseten asked these questions. Check each question for answer access.

Other topics for Baseten preparation

Explore the other topics mapped to Baseten in our preparation guide.

Continue your Baseten interview preparation

Continue with related questions and the concept curriculum. Free questions are available in every track; check pricing for full access.

Independent and not affiliated with Baseten. All trademarks belong to their owners.