Staff Software Engineer GPU Inference
Summary
Staff engineer who builds and operates Cerebras' GPU inference stack — serving workers, runtimes, GPU nodes, networking, and rack-scale infrastructure — improving reliability, latency, throughput, and numerical correctness. Core tech: C++, Python, GPU serving frameworks (vLLM/SGLang/TensorRT-LLM/Triton), Linux, Kubernetes, and profiling/observability tooling.
You will build and operate the GPU prefill path across APIs, model-serving workers, runtimes, GPU nodes, networking, and rack-scale infrastructure. You will improve operational readiness, reliability, numerical correctness, performance, and capacity efficiency while developing benchmarks, profiling automation, dashboards, and regression gates.
Responsibilities
- Design, build, deploy, and maintain the GPU inference stack
- Establish deployment, upgrade, rollback, health-checking, capacity-management, and failure-recovery practices
- Define service-level indicators and objectives for GPU-backed inference
- Optimize time to first token, throughput, latency, GPU utilization, memory efficiency, and capacity
- Improve scheduling, batching, caching, parallelism, admission, quantization, and graph execution
- Debug failures and performance regressions across application, runtime, distributed systems, and hardware layers
- Build validation and regression infrastructure for model quality and numerical correctness
- Develop benchmarks, workload replay tools, profiling automation, dashboards, and regression gates
Requirements
- 8+ years of software engineering experience
- Experience building, operating, or optimizing production inference systems for demanding GPU workloads
- Programming experience in C++ and Python
- Experience with a model-serving framework such as vLLM, SGLang, TensorRT-LLM, or Triton Inference Server
- Understanding of GPU execution and performance
- Experience debugging distributed systems across multiple layers
- Experience with Linux, containers, Kubernetes, observability, CI/CD, and latency-sensitive production services
- Ability to design benchmarks and translate performance findings into production improvements
- Communication and technical leadership skills