Lakshya

Artificial Intelligence

Frontier track Small · most durable

5 · AI Platform & Inference Infrastructure Engineer

Where the GPUs, the throughput and the cost-per-token live. Fewer postings, less competition per posting.

What the job actually is

Making training and inference fast and affordable at scale. Cluster scheduling, distributed training that survives node failure, serving stacks, batching strategy, quantisation, and the unglamorous arithmetic of how many tokens per second you get per dollar. This is the archetype whose skills age best — the specific models change yearly, the systems constraints do not.

Titles this hides behind

Platform Engineer — AI Infrastructure Software Engineer, AI PlatformPerformance Engineer, Inference ML Engineer (Training Infra)Performance Engineer, On-Device Inference ML Ops Engineer

Who is hiring it

Baseten, Modal, Together, Fireworks, Databricks, Snowflake, plus the infrastructure organisation inside every frontier lab. In India, Sarvam is hiring both training infrastructure and inference performance, and Glance is hiring on-device inference — which is a distinctly Indian specialism driven by a mobile-first market.

How to recognise it in a JD

Look for throughput, latency, quantisation, KV cache, vLLM, Triton, CUDA, FSDP, multi-node, cost per token. These terms are rare corpus-wide — CUDA 4%, distributed training 6% — which is precisely why they are a clean signal when present.

The loop

Systems coding screen Concurrency, memory, profiling. Closer to a classical backend screen than an ML one.
Distributed systems design — the decisive round Design a serving stack to a latency and cost target. Expect to be pushed on the arithmetic until it either holds or does not.
Performance deep dive A time you made something faster. They will ask how you measured, before and after.
Behavioural / on-call reality On-call appears in only 5% of AI JDs corpus-wide, but disproportionately in this archetype. Ask about rotation.

Prepare like this

  • Do the memory arithmetic until it is instant — parameters, optimiser state, activations, and KV cache growth as a function of batch and sequence length. This gets asked directly and answered badly.
  • Know continuous batching, paged attention, speculative decoding, and tensor versus pipeline parallelism well enough to explain the trade-off, not just the name.
  • Bring one measured win. A latency or cost-per-token improvement with a before, an after, and the profiling that found it.
  • Serve a real model yourself on rented GPUs and hit an actual throughput target. The gap between reading and doing shows immediately here.

Be ready for

  • "Estimate the GPU memory to serve a 70B model at batch 32, 8k context." Arithmetic, out loud, with your assumptions stated.
  • "p50 latency is fine, p99 is terrible. Where do you look?" Queueing, batching, preemption, and cold starts.
  • "Cut inference cost 40% without hurting quality. What do you try, in what order?"

Red flags in the posting

"AI Platform" that is actually Kubernetes. Some postings under this title are general platform engineering with no model-specific work at all. Ask in the screen whether the team owns the serving stack or merely the cluster it runs on.

Compensation

United States — base$200k – $350k Scarcity-priced. The corpus 90th percentile of $350k is populated heavily by this archetype and archetype 4.
India — total₹35L – ₹75L Estimated. Very few domestic postings, so the range is wide and weakly supported.

What this role tests

Themes, and where to learn them

These chapters are shared across every role that tests them, so preparation here compounds rather than being spent once.

Design under constraintThe decisive round for SRE, AI Platform, Infrastructure — and the reason strong engineers fail it.READ THE CHAPTER →Latency and cost arithmeticThe one piece of maths that is genuinely counterintuitive — and it comes up in every agent interview.READ THE CHAPTER →

Practice questions across all themes →  ·  Back to Artificial Intelligence →