Staff Software Engineer - AI Compute Libraries & Performance

📋 Description

• Build high-performance AI compute libraries for next-generation AI hardware.
• Work on linear algebra, tensor operations, numerical correctness, and low-level performance
• Design and implement kernels for linear algebra and tensor ops (GEMM, batched GEMM, convolutions
• Profile for performance, optimise memory layout, threading, and kernel launch efficiency.
• Mentor teammates, provide technical leadership, and help steer engineering decisions.

🎯 Requirements

• Excellent programming in C++ and Python.
• Strong experience with linear algebra and tensor operations; kernel development for AI compute.
• Hands-on with profiling, performance tuning, and numerical stability across FP32/FP16/BF16/FP8.
• Experience with PyTorch integration or similar custom ops/extensions (dispatch keys).
• Ability to operate at Staff level, balancing hands-on work with broader technical ownership.

🎁 Benefits

• Competitive salary and annual leave policy.
• Medical and dental health plans, gym card, pension (matched up to 4%).
• Inclusive, flexible interview process and reasonable adjustments on request.