S

Sarvam

Full-stack generative AI platform for enterprises

Performance Engineer - Inference

Applied Machine LearningConfirmed live in the last 24 hours
Full-timeNo salary listed

About the Job

Requirements

  • At least 5 years of experience in machine-learning systems, including at least 2 years of production-scale inference-serving experience, with concrete outcomes such as tokens served per day, throughput improvements, or p99 latency reductions.
  • Experience serving models with 100 billion or more parameters in production across multi-node tensor, pipeline, or expert parallelism.
  • Source-level fluency in at least one of SGLang, vLLM, NVIDIA Dynamo, or TensorRT-LLM, including modifying its scheduler, KV allocator, or disaggregation path, and reading-level familiarity with the other three.
  • Experience operating and extending a distributed-serving stack, including disaggregated prefill-decode, distributed KV/cache transfer (such as Mooncake or equivalent), and cross-node routing and scheduling; production operation and modification of such a stack.
  • Ability to build and train speculative-decoding draft models or speculators, distill them from a target model, measure and tune acceptance rate against a real serving distribution, and compose speculation with the rest of the serving stack.
  • Deep understanding of KV-cache internals, including block tables, copy-on-write, prefix sharing, and fragmentation.
  • Working command of tensor parallelism, pipeline parallelism, expert parallelism, NCCL primitives, and their interaction with the scheduler.
  • Experience with multi-tenant serving, including model co-location and MIG/MPS isolation.
  • Profiling fluency with Nsight Systems, framework tracing, and py-spy or perf.
  • Read-and-modify-level proficiency in C++ and CUDA.
  • On-call ownership of an inference service-level objective.

Responsibilities

  • Own the production serving path for large distributed models end to end.
  • Read and modify the serving runtime when its standard behavior does not fit the workloads, and operate and extend a distributed-serving stack.
  • Integrate artifacts from model and kernel teams into a running multi-node, multi-tenant stack.
  • Build and train draft models or speculators, distill them from target models, and tune acceptance rates against the live serving distribution.
  • Produce and defend the latency and throughput measurements used for company planning.
  • Work across teams on architecture co-design with the kernels team, model team, and site reliability engineering organization.

Desired Qualifications

  • Upstream contributions to SGLang, vLLM, Dynamo, llm-d, TensorRT-LLM, or LMDeploy on non-trivial code paths.
  • Direct production experience with Dynamo or llm-d at scale.
  • Experience operating a forked runtime in production.
  • Published or shipped speculator work, such as a draft model or speculative-decoding technique that was trained and measured.
  • Experience with Mixture-of-Experts serving at scale, long-context workloads of 128K or more, multi-model serving, or Indic and multilingual workloads.

About the Company

Sarvam

Full-stack generative AI platform for enterprises

Sarvam AI builds a full-stack Generative AI platform and research-informed models for enterprise use in India. It combines training of custom large language models and bespoke enterprise models with an enterprise-grade platform for authoring, deployment, and distribution of AI applications. The product works by offering end-to-end tooling: researchers develop and fine-tune models, then enterprises author, deploy, and manage these models within a scalable platform that supports multilingual and diverse Indian business needs. Sarvam AI differentiates itself through its focus on the Indian market, tailoring AI solutions to linguistic diversity and enterprise requirements, and by pairing model development with deployment infrastructure to deliver cost-effective, robust performance for customer deployments. Its goal is to accelerate the adoption of Generative AI in India by making development, deployment, and distribution of AI applications more efficient and affordable for enterprises.

Company size
51-200
Company stage
Series B
Total funding
$275M
Headquarters
Bengaluru, India
Founded
2023

Recently Posted Jobs

Explore more opportunities at Sarvam

See more jobs