--- title: Software Engineer Inference Performance at Baseten description: Baseten is hiring for the Software Engineer Inference Performance role in San Francisco, CA. See the full description and apply. type: job url: https://www.foundrole.com/jobs/software-engineer-inference-performance-at-baseten-01a10e6c-b7db-7aef-b49e-72bb6c2e28e1 date: 2026-10-06T00:19:14Z og_description: Join Baseten as Software Engineer Inference Performance in San Francisco, CA. Pays $180K–$360K per year. Full-time, hybrid role. og_image: https://www.foundrole.com/og/ns94l7.png breadcrumbs: - label: Home url: https://www.foundrole.com/ - label: Search url: https://www.foundrole.com/jobs --- | | | |---|---| | **Company** | Baseten | | **Location** | San Francisco, CA (Hybrid) | | **Salary** | $180K/yr - $360K/yr | | **Type** | Full Time | | **Posted** | Oct 05, 2026 | ## Description **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. - Bring up and tune new model architectures on new hardware quickly, often in the same week they're released. - Build benchmarking frameworks that measure real-world performance across model architectures, batch sizes, sequence lengths, and hardware configurations. - Contribute upstream to open-source inference engines (vLLM, SGLang, TensorRT-LLM), and partner closely with model, infrastructure, and customer-facing teams to ship wins.   **REQUIREMENTS** - Bachelor's, Master's, or Ph.D. degree in Computer Science, Engineering, Mathematics, or related field. - Experience with one or more general-purpose programming languages, such as Python or C++. - Familiarity with LLM optimization techniques (e.g., quantization, speculative decoding, continuous batching). - Strong familiarity with ML libraries, especially PyTorch, TensorRT, or TensorRT-LLM. - Demonstrated interest and experience in LLMs. - Deep understanding of GPU architecture.   **NICE TO HAVE** - Proficiency in enhancing the performance of software systems, particularly in the context of large language models (LLMs) - Contributed to vLLM, SGLang, TensorRT-LLM, or another inference engine. - Worked on large-scale distributed serving: autoscaling, load balancing, multi-region or multi-cloud capacity. - Written or optimized GPU kernels (CUDA, Triton, CUTLASS, or similar) - Worked on quantization (FP8/FP4, AWQ, GPTQ) or speculative decoding in production. - Deep understanding of software engineering principles and a proven track record of developing and deploying AI/ML inference solutions. **BENEFITS** - Competitive compensation, including meaningful equity - (U.S. only) 100% coverage of medical, dental, and vision insurance for employee and dependents - Flexible PTO policy including company wide Winter Break (our offices are closed from Christmas Eve to New Year's Day!) - Paid parental leave - Fertility and family-building stipend through Carrot - (U.S. only) Company-facilitated 401(k) - Exposure to a variety of ML startups, offering unparalleled learning and networking opportunities. **Apply now** to embark on a rewarding journey in shaping the future of AI! If you are a motivated individual with a passion for machine learning and a desire to be part of a collaborative and forward-thinking team, we would love to hear from you. *At Baseten, we are committed to fostering a diverse and inclusive workplace. We provide equal employment opportunities to all employees and applicants without regard to race, color, religion, gender, sexual orientation, gender identity or expression, national origin, age, genetic information, disability, or veteran status.* *We are an Equal Opportunity Employer and will consider qualified applicants with criminal histories in a manner consistent with applicable law (by example, the requirements of the San Francisco Fair Chance Ordinance, where applicable).* ## Skills - Blogs - KV Cache - Machine Learning Model Training - Autoscaling - Open-Source Software - Investigation - NVIDIA Triton Inference Server - Structured Output (Llm) - Performance Optimization - Caching - Python - Linux Kernel Development - Algorithms - Distributed Tracing - SGLang - GPU Kernel Development - TensorRT-LLM - Multi-Cloud - Regression Analysis - Hardware Bring-up - Tokenization - Benchmarking - Quantization - Mathematics - AI Inference - C++ - GPU (Graphics Processing Unit) - Large Language Models (Llm) - NVIDIA CUDA - Computer Science - NVIDIA TensorRT - Inference Engine - GPU Architecture - Speculative Decoding - Machine Learning - Generalized Linear Models (Glm) - Load Balancing - vLLM - LoRa ## Benefits - Health Insurance - Dental Insurance - Vision Insurance - Parental Leave - Paid Time Off (Pto) - 401(k) Plans ## Related - [Technology & Software jobs](https://www.foundrole.com/careers/technology-and-software?utm_source=ai_markdown)- [Software Development jobs](https://www.foundrole.com/careers/technology-and-software/software-development?utm_source=ai_markdown)- [Software Developer / Engineer jobs](https://www.foundrole.com/careers/technology-and-software/software-development/software-developer-engineer?utm_source=ai_markdown)- [Browse all companies](https://www.foundrole.com/companies?utm_source=ai_markdown) ## Explore this job market - [California](https://www.foundrole.com/locations/us/california?utm_source=ai_markdown) — 264806 jobs - [Software Products](https://www.foundrole.com/sectors/technology/software-products?utm_source=ai_markdown) — 117493 jobs - [San Francisco, CA](https://www.foundrole.com/locations/us/california/san-francisco?utm_source=ai_markdown) — 26836 jobs ## Related pages - [California](https://www.foundrole.com/locations/us/california?utm_source=ai_markdown) — 264806 jobs - [Software Products](https://www.foundrole.com/sectors/technology/software-products?utm_source=ai_markdown) — 117493 jobs - [San Francisco, CA](https://www.foundrole.com/locations/us/california/san-francisco?utm_source=ai_markdown) — 26836 jobs ## Browse - [Companies](https://www.foundrole.com/companies?utm_source=ai_markdown) - [Jobs by country](https://www.foundrole.com/locations?utm_source=ai_markdown) - [Jobs in the United States by state and city](https://www.foundrole.com/locations/us?utm_source=ai_markdown) - [H1B sponsors by location](https://www.foundrole.com/h1b-sponsors?utm_source=ai_markdown) - [H1B salaries](https://www.foundrole.com/h1b-salaries?utm_source=ai_markdown) - [Sectors and industries](https://www.foundrole.com/sectors?utm_source=ai_markdown) - [Jobs by career field](https://www.foundrole.com/careers?utm_source=ai_markdown) - [Latest jobs](https://www.foundrole.com/jobs/latest?utm_source=ai_markdown)