Staff Software Engineer, Inference API
Cerebras is hiring a Staff Software Engineer to build and evolve the ML inference API layer for its disaggregated serving system, which pairs GPU-accelerated prefill with ultra-fast decode on its wafer-scale AI chip. Day to day: designing production chat/completions APIs, integrating models and runtimes (vLLM, PyTorch, ROCm), and improving latency, reliability, and observability across distributed
- Compensation
- Not disclosed
- Location
- Toronto, Canada
- Work setup
- Not specified
- Seniority
- Staff
- Employment
- Not specified
- Experience
- 5+ years
- Category
- Software engineering
- Posted
Skills
Must have
- 5+ years of software engineering experience with substantial IC ownership of production software or distributed systems
- Strong programming ability in Python and experience with performance-sensitive or highly concurrent services in C++, Go, or a similar systems language
- Hands-on experience with a model-serving framework such as vLLM, SGLang, TensorRT-LLM, Triton Inference Server, Hugging Face TGI, or equivalent
- Understanding of modern LLM inference concepts: tokenization, prompt formatting, sampling, streaming generation, continuous batching, KV-cache management, model configuration
- Experience integrating software across service, framework, runtime, and infrastructure boundaries
- Experience building stable APIs with validation, error handling, observability, compatibility, and versioning practices
- Experience with Linux, containers, Kubernetes or comparable orchestration, CI/CD, and operating latency-sensitive services in production
- Ability to diagnose correctness, reliability, and performance issues across multiple components of a distributed serving system
- Strong communication and cross-functional execution skills
- Bachelor's degree in Computer Science, Computer Engineering, Electrical Engineering, or related discipline, or equivalent practical experience
Full description
Cerebras Systems builds the world's largest AI chip, 56 times larger than GPUs. This architecture allows Cerebras to deliver industry-leading training and inference speeds; over 10 times faster than GPU-based hyperscale cloud inference services. This order of magnitude increase in speed is transforming the user experience of AI applications, unlocking real-time iteration and increasing intelligence via additional agentic computation. Cerebras works with the leading model labs, global enterprises, and cutting-edge AI-native startups. OpenAI recently announced a multi‑year partnership https://openai.com/index/cerebras-partnership/ with Cerebras, to deploy 750 megawatts of scale, transforming key workloads with ultra high-speed inference.
About the Role Cerebras is building a new generation of disaggregated AI inference systems that combine GPU-accelerated prefill with ultra-fast decode on the Cerebras Wafer-Scale Engine. We are hiring a Software Engineer to build and evolve the ML API layer that makes this heterogeneous serving system accessible, reliable, and easy to use. You will work across our inference APIs, model integration layer, request‑routing services, vLLM‑based GPU runtime, and Cerebras inference platform to deliver a consistent experience across models and accelerator backends. This role sits at the intersection of machine learning systems, API design, model serving, and distributed systems. You will enable new model architectures and inference capabilities, define stable user‑facing behavior, and ensure that features such as streaming, sampling, tool use, structured outputs, multimodal inputs, and model configuration behave correctly and consistently in production. You will work closely with model enablement, compiler, runtime, cloud infrastructure, product, and customer‑facing teams. This is a hands‑on software engineering role for someone who enjoys turning rapidly evolving ML capabilities into durable, production‑quality APIs.
Responsibilities
- Build production ML inference APIs.
- Design, implement, and maintain APIs for chat completions, text generation, streaming, model configuration, tool calling, structured outputs, multimodal inputs, and other emerging inference capabilities.
- Deliver a unified serving experience. Create consistent request and response semantics across GPU prefill, Cerebras decode, and other heterogeneous inference backends.
- Enable new models and capabilities. Integrate emerging foundation models, tokenizers, prompt formats, sampling methods, attention variants, multimodal inputs, and model‑specific features into the serving platform.
- Own API compatibility and evolution. Maintain compatibility with widely adopted inference interfaces while designing Cerebras‑specific extensions. Establish clear versioning, deprecation, validation, and backward‑compatibility practices.
- Integrate with model‑serving runtimes. Extend and integrate custom inference services with vLLM, PyTorch, Hugging Face libraries, the AMD ROCm stack, and Cerebras runtime components.