AI Infra Interviews logo
CORE AI INFRASTRUCTURE

NVIDIA AI Infrastructure Engineer interview questions

NVIDIA hires across every layer this site covers: deep learning library and performance engineers (cuDNN, CUTLASS, TensorRT, TensorRT-LLM, LLM performance), developer technology engineers, Triton Inference Server systems engineers, DGX Cloud performance engineers who benchmark distributed systems to define cluster architecture, and GPU and HPC infrastructure engineers who run NVIDIA's own fleets. The loop is domain-heavy and C++-first: a recruiter call, one or two 45 to 60 minute phone screens that mix C++ fundamentals (virtual dispatch, memory model, move semantics) with graph and array problems, then four to six onsite rounds where the team's domain dominates; candidates consistently report that domain expertise in the team's area is the filter. Reported kernel questions include memory coalescing, shared-memory bank conflicts, warp divergence, parallel reduction, GEMM tiling, and profiling with Nsight. Loops run four to eight weeks. NVIDIA India hires in Pune and Bengaluru.

CHIP AND PLATFORM VENDORS

They build the silicon and the software that drives it, so depth in the team's own domain outranks breadth almost everywhere.

Loop leans on: Microarchitecture, kernels, compilers, interconnect, benchmarking. Compare the other chip and platform vendors

The NVIDIA AI Infrastructure Engineer interview process

Documented

How the NVIDIA AI Infrastructure Engineer interview experience actually runs — the rounds, what each stage tests, and the signals candidates report. Last reviewed September 4, 2026.

RoleDeep Learning Software Engineer / GPU and HPC Infrastructure Engineer / DGX Cloud Performance EngineerLoop4 to 8 weeks typical; long loops are a recurring complaintAI toolsNo first-party statement found; no candidate report of AI being permitted.
  1. 1
    Recruiter callTeam and level.
  2. 2
    Phone screensOne or two rounds of 45 to 60 minutes: resume discussion plus C++ fundamentals (virtual dispatch, memory model, move semantics) and graph or array problems.
  3. 3
    Take-home (team-dependent)One 2022 report of two 48-hour exercises; most 2023 to 2026 reports do not mention one.
  4. 4
    OnsiteFour to six rounds of about an hour with domain-specific content: systems architecture, performance, team-specific coding (whiteboard CUDA and kernel pseudo-code reported for DevTech and library loops), and a behavioral round; sometimes a hiring-manager or director round.
WHAT THEY'RE EVALUATING
  • Expertise in the team's domain, reported as the dominant filter
  • C++ first, CUDA, Python for tooling
  • Memory coalescing, bank conflicts, warp divergence, parallel reduction, GEMM tiling, Nsight profiling (reported kernel questions)
  • TensorRT-LLM, vLLM, SGLang and speculative decoding for LLM performance roles

Specific kernel questions are mostly aggregator-reported with few dated first-hand debriefs; treat individual items as partial.

Compiled from our research and publicly available information (candidate reports and company interview guides). Interview loops change and are continuously iterated, and they vary by team, level, and region. Treat this as directional preparation, not an official spec, and confirm the exact rounds with your recruiter or hiring point of contact.

NVIDIA AI Infrastructure Engineer salary

What we can trace, labelled by where it came from. We publish a band only where there is a source behind it, so some of this page is a gap rather than a number.

REPORTED FOR NVIDIA
$184K - $288KbaseEmployer posting

This band covers the title DGX Cloud Performance Engineer (L4, new grad MS/PhD). A band belongs to a title, not to a company, and attaching one to the wrong title is the most common error in published AI infra compensation data.

2026 posting; the L3 band was $148K to $235,750 and the GPU-clusters infra L2 band $120K to $189,750.

HIRING FROM INDIA
Multinational with an India engineering centre

An established India presence, usually Bengaluru, Hyderabad or Pune, hiring on a local band with the parent company's level structure. Far more attainable than the global-remote route, with listed-company equity and the usual multinational benefits.

LEVELREPORTED FOR THIS EMPLOYER TYPE
Early career (IC1-IC2 equivalent)₹26 LPA - ₹45 LPA
Senior (IC3 equivalent)₹37 LPA - ₹85 LPA
Staff and above (IC4+ equivalent)₹69 LPA - ₹1.4 Cr

Reported total compensation for NVIDIA software engineers in India by level, per levels.fyi self-reports (accessed September 2026; IC3 median about ₹62 LPA, IC4 median about ₹94 LPA), used as the reference for this employer type. Not a figure reported for this company or for this exact title; bands vary by internal level and by company.

Full method, US bands by level, and the three India tiers side by side are in the AI infra salary guide, including what actually moves your number between these tiers.

Questions modeled on NVIDIA loops

107 questions · 42 unlocked for you

More from the tracks NVIDIA's loop tests

The highest-signal questions across NVIDIA's core tracks.

8 questions · 1 unlocked for you

Go deeper on the topics NVIDIA's loop tests

The tracks that map to a NVIDIA AI Infrastructure Engineer loop, ordered easy to hard.

The concepts NVIDIA's AI Infrastructure Engineer loop assumes you know

The vocabulary and mental models behind NVIDIA's questions, from our curriculum. Start with the foundations free; the deeper, interview-defining ideas are part of premium.

KERNELS & COMPILERS

Foundational
CUDA Programming ModelCUDA splits a program into a host that allocates, copies and enqueues work, and a device that runs thousands of identical threads organized as a grid of blocks. Getting the split right, and knowing that a launch returns before the kernel runs, decides whether your first live-coding kernel produces a correct number or a silent zero.
CoreSign in
Memory CoalescingA warp's 32 threads issue one memory request together, and the hardware serves it in 32-byte sectors. Coalescing is arranging addresses so those sectors are full of bytes the warp will use. It decides whether a bandwidth-bound kernel moves at the HBM rate or at an eighth of it, and it is the pattern NVIDIA's trace-classification interview question tests.
Advanced🔒 Premium
Shared Memory and Bank ConflictsShared memory is the programmer-managed SRAM inside each SM, split into 32 four-byte banks that serve one word each per cycle. When several lanes of a warp hit the same bank at different addresses the access serializes, and a 32-way conflict makes a shared-memory-bound loop run over ten times slower. Padding, XOR swizzles, cp.async and TMA are the tools that decide whether a tiled kernel gets the bandwidth it staged data for.
Advanced🔒 Premium
Occupancy and Register PressureOccupancy is the fraction of an SM's 64 warp slots that are resident, and it is capped by the 65,536 registers and 228 KB of shared memory each block consumes. It decides how much memory latency the hardware can hide for free, but the fastest kernels on a GPU routinely run at 25 percent, so the interview skill is knowing when to raise it and when to stop.

GPU & ACCELERATOR ARCHITECTURE

Foundational
GPU Execution ModelA GPU hides memory latency with parallelism instead of caches: thousands of threads in flight, scheduled in warps of 32, pinned to streaming multiprocessors that switch between warps for free whenever one stalls. Every performance conversation in an AI infra loop, from occupancy to why decode is slow, rests on this one mechanism.
Foundational
GPU Memory HierarchyA GPU has four places a byte can live, and they differ by a thousandfold in bandwidth: registers, shared memory on the SM, a chip-wide L2, and HBM off-chip. Almost every kernel optimization is a decision about which level a value is read from and how many times. Knowing the sizes and bandwidths for an H100 cold is what lets you say why a kernel is slow before you profile it.
CoreSign in
Tensor Cores and Matrix UnitsTensor cores are fixed-function units that compute a small matrix multiply-accumulate per instruction, and they are where almost all of a modern GPU's FLOPS live: 989 dense bf16 TFLOPS on an H100 against about 67 from the general-purpose lanes. Only dense, well-shaped matrix multiplication at a supported precision can use them, which is why GEMMs reach peak and nothing else does, and why precision choices are throughput choices.
Advanced🔒 Premium
Memory-Bound vs Compute-Bound KernelsEvery kernel is limited by one of two walls: how fast bytes arrive from HBM, or how fast the tensor cores can multiply. Which wall applies is decided by arithmetic intensity against the ridge point, and the two regimes need opposite fixes. Decode, LayerNorm and softmax are memory-bound; prefill GEMMs are compute-bound; the interview question is which one you are looking at and what you would do about it.

INFERENCE & SERVING

Foundational
Prefill vs DecodeAn LLM request runs in two phases with opposite hardware profiles: prefill reads the whole prompt in one compute-bound pass and decides time to first token, decode emits one token per forward pass and is bound by memory bandwidth. Every serving decision, from batch size to which GPU to buy to whether to split the two phases across machines, follows from that split.
Foundational
The KV CacheThe KV cache stores each token's attention keys and values so decode never recomputes them, turning a quadratic cost into a linear one at the price of memory that grows with every token in every concurrent sequence. Its size, 128 KB per token for Llama 3.1 8B and 320 KB for 70B in bf16, is what caps concurrency and context on a given GPU, so it decides batch size, replica count and whether a model fits at all.
CoreSign in
Continuous BatchingContinuous batching schedules at the granularity of a single decode step instead of a whole request, so a finished sequence's slot is refilled on the next iteration rather than when the longest request in the batch ends. It is the scheduling idea that turned LLM serving from a padded, half-idle GPU into one that stays full, and it decides how the engine's scheduler, memory manager and latency SLOs interact.
Advanced🔒 Premium
PagedAttentionPagedAttention stores the KV cache in fixed-size blocks scattered across HBM and maps each sequence's logical positions to physical blocks through a block table, the same trick an operating system uses for virtual memory. It removes the reservation and fragmentation waste of contiguous allocation, lets blocks be shared between sequences, and is why an engine can decide admission by counting free blocks.

CODING FOR INFRA

Foundational
The GPU Credit Scheduler PatternThe most widely reported coding problem in AI infrastructure loops is a small scheduler: accounts hold credits, jobs arrive with a cost and a priority, and you must decide which jobs run, in what order, without letting any account overspend, then extend it under follow-ups (refunds, reservations, concurrency limits, fairness). It is not a trick question; it is a test of whether you can model state cleanly, pick the right data structures, keep invariants under mutation, and talk about complexity while typing. This page works the problem from the first line to the fourth follow-up, with the code, the invariants, and the derivations.
CoreSign in
Rate-Limiting AlgorithmsA rate limiter answers one question, 'may this request proceed now?', and the three classic algorithms answer it with different shapes of fairness and memory: the token bucket allows bursts up to a capacity and refills at a rate, the leaky bucket smooths output to a fixed rate, and sliding windows count recent requests exactly or approximately. AI platforms limit in tokens as well as requests, per tenant, across many gateways, which adds two twists: a request's cost is unknown until it finishes, and the counters must be shared. This page derives each algorithm, implements the token bucket correctly, and covers both twists.
Advanced🔒 Premium
Batching Queues and BackpressureWrite a request batcher is the coding round's version of the serving engine's scheduler: requests arrive one at a time, the GPU wants them in groups, and the batcher decides when a group is full enough to send without holding anyone too long or accepting more than it can hold. The two knobs are the maximum batch size and the maximum wait, the invariant is a bounded queue, and the follow-ups (priorities, cost-aware batching, cancellation, bounded in-flight batches) are the ideas the real engines carry. This page implements the batcher in asyncio, derives what each knob buys, and walks the follow-ups.
Advanced🔒 Premium
Interval Merging and Utilization LogsGiven busy intervals per GPU, when was the whole cluster idle? What was the utilization per hour from a log of start and stop events? Which jobs overlapped? These are the interval problems of the infrastructure coding screen, and they share one tool: sort the endpoints and sweep. The sweep line turns every variant into a single pass with a counter, the sort is the only thing that costs more than linear time, and the edge cases (touching intervals, zero-length events, an unterminated start) are where candidates lose the round. This page works the standard problem and its relatives with code, tests and the complexity derivation.

NETWORKING & STORAGE

Foundational
NCCL and Collective AlgorithmsNCCL is the library every PyTorch collective lands in, and its choice of ring or tree, channel count and protocol decides whether an all-reduce runs at fabric speed or at a third of it. Knowing what NCCL_DEBUG=INFO prints, and which environment variable changes which decision, is the difference between tuning a cluster and guessing at it.
CoreSign in
RDMA, InfiniBand and RoCEv2Training across nodes moves hundreds of gigabytes per step, and a CPU-driven TCP stack cannot feed a 400 Gb/s link. RDMA lets a NIC write straight into a remote GPU's memory with no kernel and no copies, and it runs over two fabrics: InfiniBand, which is lossless by design, and RoCEv2, which is Ethernet made lossless by configuration. The choice is operational as much as technical, and the numbers that decide it are per-GPU bandwidth, the collective's volume, and who will debug a pause storm at 3 a.m.
Advanced🔒 Premium
Rail-Optimized and Fat-Tree FabricsA GPU cluster's network is built from two ideas: a fat tree (Clos) that gives every node a path to every other node with a chosen amount of oversubscription, and rail optimization, which wires GPU i of every node to the same leaf switch so the collectives that dominate training stay one hop away. Sizing one is arithmetic on port counts, and the interview question is usually that arithmetic: how many switches, what oversubscription, and where the NVLink domain ends and the fabric begins.
Advanced🔒 Premium
Congestion Control for AI FabricsCollective traffic is the worst case a network can see: hundreds of senders transmit to the same receiver at the same instant (incast), every flow is large and long-lived, and RDMA cannot tolerate a dropped packet. Congestion control is the set of mechanisms (PFC, ECN with DCQCN, adaptive routing, packet spraying) that keep queues from overflowing without stalling the fabric. On plain Ethernet a busy all-reduce can fall to about 60% of link rate; with a tuned control loop it holds above 90%. Reading the counters that show which one you have is the on-call skill.

Where to apply, and official NVIDIA resources

Straight from NVIDIA: open roles and the company's own hiring guidance. Prep here, then apply there.

External links to NVIDIA's own pages. Roles and processes change; always confirm on the official site.

ABOUT THE ROLE
NVIDIA INTERVIEW FAQ
What is the NVIDIA AI Infrastructure Engineer interview process?

Deep Learning Software Engineer / GPU and HPC Infrastructure Engineer / DGX Cloud Performance Engineer. Typical loop: 4 to 8 weeks typical; long loops are a recurring complaint. Stages: Recruiter call → Phone screens → Take-home (team-dependent) → Onsite. Key focus: Expertise in the team's domain, reported as the dominant filter. Compiled from public reports; loops change over time, so confirm the exact rounds with your recruiter.

Does NVIDIA hire AI infrastructure engineers?
What does the NVIDIA AI infrastructure interview test?
What CUDA questions does NVIDIA ask?
What is the NVIDIA AI infrastructure engineer salary?
How long is the NVIDIA loop?

Walk into your NVIDIA AI Infrastructure Engineer interview ready

Unlock every AI infra interview answer, ordered easy to hard, plus the full concept curriculum, for 6 months. One payment, no auto-renewal. Free questions and concepts in each track, no card needed to start.

Or create a free account to unlock more free answers per topic.

Other AI Infrastructure Engineer interviews to prep

Companies whose loops test the same tracks as NVIDIA's.

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