C

Callosum

Orchestrates AI workloads across specialized silicon

Hardware-Aware Model Optimisation - Member of Technical Staff

Natural Language Processing & LLMsPosted on Nov 13
Full-timeLondon, UK

About the Job

Requirements

  • Deep understanding of modern large language model architectures, including attention mechanisms, model structure, and their implications for inference.
  • Experience in model architecture research, efficient inference, or hardware-aware machine learning.
  • Familiarity with architectures such as mixture-of-experts, diffusion, JEPA, or alternative attention mechanisms.
  • Strong intuition for interactions between model capability, numerical behaviour, memory requirements, and execution efficiency.
  • Experience designing rigorous experiments to assess model quality, efficiency, and the effects of architectural modifications.

Responsibilities

  • Develop hardware-aware transformations that adapt models to different accelerator architectures and execution environments.
  • Explore numerical, structural, and memory-related optimizations to create model variants with different capability, latency, and cost trade-offs.
  • Investigate new inference algorithms, including speculative decoding, designed to exploit heterogeneous computational resources.
  • Develop automated methods for discovering, characterizing, and optimizing hardware-specialized model variants.
  • Build evaluation methods that measure model quality and establish the trade-offs introduced by hardware-aware adaptations.

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