Member of Technical Staff - Inference Infrastructure

Causal Labs

$130K — $180K *
Information Technology
Less than 5 years of experience
Job Overview by Ladders

Qualifications

  • 5-7 years experience optimizing inference and serving systems for performance
  • Solid understanding of distributed computing and GPU optimization
  • Proficient in deep learning frameworks like PyTorch and JAX
  • Strong coding skills with a focus on performance and maintainability
  • Contributions to open-source projects in the inference or systems domain are a plus

Responsibilities

  • Build high-throughput inference systems for evaluation and backtesting
  • Design techniques to enhance real-time inference efficiency
  • Optimize the inference stack to maximize hardware performance
  • Extend orchestration tools for distributed inference tasks
  • Establish reliability and observability standards across the stack
  • Collaborate with researchers for high-performance inference solutions

Benefits

  • Flexible work arrangements
  • Opportunities for professional development and training
  • Access to cutting-edge technology and tools
  • Collaborative and innovative work culture
Full Job Description
Responsibilities

Your mission is to make inference so fast and cheap that evaluation never gates research.
  • Build high-throughput inference systems for large-scale evaluation, backtesting, and scoring against historical physical observations
  • 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 (e.g. Kubernetes, Ray, Slurm) for distributed inference and large-batch evaluation sweeps
  • Establish standards for reliability, observability, and reproducibility across the inference stack, so every evaluation is trustworthy and repeatable
  • Collaborate with researchers to enable high-performance inference for novel architectures as they emerge


What we're looking for

We value a relentless approach to problem-solving, rapid execution, and the ability to quickly learn in unfamiliar domains.
  • Experience building or optimizing inference and serving systems for throughput and latency (e.g. TensorRT)
  • Understanding of distributed compute, GPU parallelism, and hardware-aware optimization
  • Deep familiarity with deep learning frameworks (e.g. PyTorch, JAX) and their underlying system architectures
  • Strong engineering skills: performant, maintainable code and the ability to debug complex codebases
  • Bonus: contributions to open-source inference or systems infrastructure (e.g. vLLM, SGLang, Triton)

Similar Jobs

More Jobs at Causal Labs

More Information Technology Jobs

Find similar Member of Technical Staff - Inference Infrastructure jobs: