Staff Engineer, Large-Scale Model Inference

SpaceX Super Intelligence; Bellevue (WA); On-site; USD 180,000 - 440,000; Full time

Benefits offered by this job

Equity

Medical, Vision, Dental coverage

401(k) retirement plan

Disability & Life Insurance

Job summary

SpaceXSI is building a high-performance inference platform to serve Grok to millions of users with speed and reliability. As a Member of Technical Staff - Inference, you will own from distributed infrastructure to deep low-level optimizations, shaping how fast users interact with Grok at massive scale.

You will work on batching, caching, load balancing, and GPU kernel optimizations, with a focus on 100% uptime and minimal tail latency. This role offers equity and comprehensive benefits.

Qualifications

• Deep low-level systems programming in C/C++ or Rust.
• Experience with large-scale, high-concurrent production serving.
• Experience with GPU inference engines (vLLM, SGLang, Triton, TensorRT-LLM, etc.).
• Strong background in system optimizations: batching, caching, load balancing, parallelism.
• Low-level inference optimizations: GPU kernels, code generation.
• Algorithmic inference optimizations: quantization, speculative decoding, distillation, low-precision numerics.
• Experience with testing, benchmarking, and reliability of inference services.
• Experience designing and implementing CI/CD infrastructure for inference.

Responsibilities

• Architect and implement scalable distributed infrastructure for model serving (load balancing, auto-scaling, batch scheduling, global KV cache).
• Optimize latency and throughput of model inference under real production workloads.
• Build reliable, high-concurrency serving systems that serve billions of users with 100% uptime, 0% error rate, and excellent tail latency.
• Benchmark, fine-tune, and accelerate inference engines (including low-level GPU kernel work and code generation).
• Develop custom tools to trace, replay, and fix issues across the full stack — from orchestration down to GPU kernels.
• Create robust CI/CD infrastructure for seamless endpoint deployment, image publishing, and inference engine updates.
• Accelerate research on scaling test-time compute, RL rollout, and model-hardware co-design for next-generation systems.

Skills

C/C++

Rust

Large-scale serving

GPU inference

Batching

Caching

Load balancing

Parallelism

GPU kernels

Code generation

Quantization

Speculative decoding

Distillation

Low-precision numerics

Testing

Benchmarking

Reliability engineering

CI/CD for inference

Job description

SpaceXSI is building a high-performance inference platform to serve Grok to millions of users with speed and reliability. As a Member of Technical Staff - Inference, you will own from distributed infrastructure to deep low-level optimizations, shaping how fast users interact with Grok at massive scale.

You will work on batching, caching, load balancing, and GPU kernel optimizations, with a focus on 100% uptime and minimal tail latency. This role offers equity and comprehensive benefits.