Causal
Member of Technical Staff — Inference Infrastructure
San Francisco. Posted 3 months ago
The role
You will build high-throughput inference systems for large-scale evaluation and backtesting. You will optimize the inference stack and extend orchestration frameworks to support distributed inference.
What you will do
- Build high-throughput inference systems for large-scale evaluation, backtesting, and scoring
- Design and implement techniques that improve latency, throughput, and efficiency for real-time inference
- Optimize the inference stack to fully utilize hardware FLOPs, bandwidth, and memory
- Extend orchestration frameworks for distributed inference and large-batch evaluation sweeps
- Establish standards for reliability, observability, and reproducibility across the inference stack
- Collaborate with researchers to enable high-performance inference for novel architectures
Required qualifications
- Experience building or optimizing inference and serving systems for throughput and latency
- Understanding of distributed compute, GPU parallelism, and hardware-aware optimization
- Deep familiarity with deep learning frameworks and their underlying system architectures
- Strong engineering skills: performant, maintainable code and the ability to debug complex codebases
Nice to have
- contributions to open-source inference or systems infrastructure (e.g. vLLM, SGLang, Triton)
About the company
Causal is building a Large Physics foundation Model to achieve general causal intelligence. The founding team has built and deployed AI at institutions like DeepMind, Waymo, Cruise, Insitro, Nabla Bio, and CERN.