Accelerator Systems Software - Member of Technical Staff

About the Company

Callosum is the Intelligent Systems Company. We believe intelligence emerges from the diversity of co-optimised mechanisms working together. We co-evolve models, workflows, and silicon to build the infrastructure required for the next era of heterogeneous AI compute.

Responsibilities

• Build and maintain kernels, device drivers, and firmware for heterogeneous accelerators
• Design and implement accelerator scheduling at the execution level, including kernel launches, dataflow optimisation, and resource management
• Optimise execution paths for latency, throughput, and resource utilisation across accelerator types
• Work closely with internal teams and hardware vendors to onboard new accelerator platforms
• Contribute to low-level runtime software that bridges the inference stack and underlying hardware

Requirements

• Deep experience in kernel development, device driver authoring, or firmware engineering
• Familiarity with accelerator programming models such as CUDA, ROCm, or vendor-specific SDKs
• Experience with compiler infrastructure like MLIR or LLVM
• Strong debugging skills in environments with limited tooling and incomplete documentation
• Demonstrable interest in a variety of accelerator microarchitectures

Benefits

• Competitive Salary
• Equity & Ownership
• Private healthcare
• Visa sponsorship and relocation benefits