Compute Library Engineer

TN United Kingdom; Cambridge; Hybrid; GBP 80,000 - 90,000; Full time

Job summary

An exciting opportunity awaits a Compute Library Engineer at a rapidly growing Semiconductor scale-up in Cambridge. This role involves developing and optimising high-performance kernels for ML operators on NPU architectures. You will have the chance to mentor junior engineers and collaborate with teams to integrate innovative solutions into the NPU framework. Join a forward-thinking company that values your ideas and offers a stable, supportive environment for career progression. If you thrive in dynamic settings and want to make an impact, this position is perfect for you.

Qualifications

• Extensive experience in kernel development projects for GPUs.
• Strong C++ development skills and understanding of ML frameworks.

Responsibilities

• Develop and optimise high-performance kernels for ML operators on NPU architectures.
• Collaborate with hardware and software teams for kernel integration.

Skills

Kernel development for GPUs

OpenCL

CUDA

Machine Learning frameworks (TensorFlow, PyTorch)

C++ development

Independent work in fast-paced environments

Job description

Compute Library Engineer

Salary: £80,000 - £90,000 + bonus & hybrid working!

I am currently working with a Cambridge-based, multinational Semiconductor scale-up focused on developing AI accelerators. This is an exciting opportunity to work in a rapidly changing environment where your innovative ideas can lead to new products, services, and customer experiences.

The company is successful and growing, offering career progression and a stable, supportive environment.

They are seeking a Compute Library Engineer to develop and optimise high-performance kernels for ML operators on NPU architectures. The ideal candidate will have experience mentoring junior engineers and working with hardware features of GPUs and accelerators designed for AI applications. Collaboration with hardware and software teams to integrate kernels into the NPU framework is also essential.

Requirements for a successful Compute Library Engineer

• Extensive experience in kernel development projects for GPUs
• Involvement in OpenCL, CUDA, or similar parallel programming languages
• Understanding of ML frameworks such as TensorFlow and PyTorch
• Strong C++ development skills
• Ability to work independently in a fast-paced environment

If you are interested in this opportunity within an exciting Semiconductor scale-up, please apply to learn more. For other software roles across the UK, please contact Jack Bird.