Software Engineer - Inference Performance

About the role

ABOUT BASETEN

Baseten powers mission-critical inference for the world's most dynamic AI companies, like Cursor, Notion, OpenEvidence, Abridge, Clay, Gamma, and Writer. By uniting applied AI research, flexible infrastructure, and seamless developer tooling, we enable companies operating at the frontier of AI to bring cutting-edge models into production. We're growing quickly and recently raised our $1.5B Series F https://www.baseten.co/blog/announcing-our-series-f/, led by Altimeter Capital, Conviction Partners, and Spark Capital. Join us and help build the platform engineers turn to ship AI products.

THE ROLE

We're looking for inference performance engineers who want to make the world's most demanding AI workloads run faster and more efficiently. You'll work across the stack, from the inference engine and runtime through scheduling, serving, and routing. Along the way you'll apply techniques like prefill/decode disaggregation, speculative decoding, and KV-cache management. You'll reason from first principles about where time and memory go, find what's holding performance back, and close the gap. Your work directly impacts how fast our customers' models run and how efficiently we serve them. This role is ideal for someone who thrives in a fast-paced startup environment and is eager to make significant contributions to the exciting field of LLM inference.

EXAMPLE INITIATIVES

You'll get to work on these types of projects as an Inference Performance engineer:

- Agentic Kernels in Production https://www.baseten.co/blog/agentic-kernels-in-production/

- How we built the new fastest API for GLM-5.2 https://www.baseten.co/blog/how-we-built-the-new-fastest-api-for-glm-52/

- Live draft model training for speculative decoding https://www.baseten.co/blog/live-draft-model-training-for-speculative-decoding/#eliminating-traditional-bottlenecks

- Making Kimi K3 Tokenization 18x faster https://www.baseten.co/blog/making-kimi-k3-tokenization-18x-faster-for-million-token-agentic-workloads/

- The Baseten Inference Stack https://www.baseten.co/resources/guide/the-baseten-inference-stack/

- Driving model performance optimization https://www.baseten.co/blog/driving-model-performance-optimization-2024-highlights/

RESPONSIBILITIES

- Implement and productionize cutting-edge inference techniques, working deep in runtime internals. That includes quantization, speculative decoding, KV-cache reuse, chunked prefill, LoRA, guided generation for structured outputs, and custom scheduling and routing algorithms.

- Profile and optimize inference end to end, from kernel launch overhead and memory layout up to request scheduling, prefill/decode disaggregation, and cache-aware routing. Run cross-layer investigations, such as tracing a tail-latency regression from request timing through routing and batching down to a kernel.

- Turn performance into cost savings. Improve tokens per GPU-hour, raise utilization, and give customers and internal teams clear latency/throughput/cost tradeoffs.