C

Callosum

Orchestrates AI workloads across specialized silicon

Inference Engine Development - Member of Technical Staff

Applied Machine LearningUpdated on Nov 14
Full-timeLondon, UKNo salary listed

About the Job

Requirements

  • Deep familiarity with SGLang, vLLM, or comparable inference frameworks and their internal execution mechanisms.
  • Strong understanding of how batching, attention implementations, caching, and scheduling interact to determine inference performance.
  • Experience with high-performance machine-learning inference systems, including Python, C++, CUDA, or similar technologies.
  • Familiarity with techniques such as PagedAttention, chunked prefill, prefix caching, or optimized attention kernels.
  • Strong systems engineering instincts and the ability to reason across inference software and hardware boundaries.

Responsibilities

  • Extend inference systems such as SGLang and vLLM, alongside developing custom inference engine technologies.
  • Develop hardware-aware execution strategies that exploit the capabilities of different accelerator architectures.
  • Optimize inference scheduling, batching, attention execution, and memory management for different workload characteristics.
  • Explore new approaches to parallelism, disaggregation, and efficient execution across heterogeneous resources.
  • Develop interfaces that expose inference engine capabilities and execution state to higher-level orchestration systems.

About the Company

Callosum

Orchestrates AI workloads across specialized silicon

Callosum builds AI infrastructure that unifies different AI models and hardware chips to run workloads on the best combination of model and chip for each task. Its Tailored Inference APIs route tasks to the optimal model–chip pair while honoring cost, energy, and latency constraints, using partners like Cerebras and Rebellions for specialized silicon. Unlike approaches that rely on one model and uniform hardware, Callosum orchestrates multiple models and chips to maximize efficiency. The company's goal is to enable scalable, cost-efficient AI inference across heterogeneous hardware by intelligently matching tasks to the right combination of model and accelerator.

Company size
11-50
Company stage
Seed
Total funding
$110.3M
Headquarters
London, United Kingdom
Founded
2025

Recently Posted Jobs

Explore more opportunities at Callosum

See more jobs