AI Kernel Engineer

fullTime Computer Occupations

You will play a key role in building and optimising high-performance AI kernels for a next-generation compute platform. This role focuses on enabling efficient execution of AI and LLM workloads by developing, profiling, and optimising kernels across varying hardware configurations.

You will be responsible for:

• Developing AI/LLM kernels and operators for efficient inference on a specialised compute platform
• Optimising kernel performance across different hardware configurations and workloads
• Profiling and analysing performance across compute, memory, and parallelism to identify bottlenecks
• Optimising low-level C/C++ code to maximise hardware utilisation
• Collaborating across the AI inference stack, including runtime, compiler, and system layers
• Contributing to improvements in toolchain, compiler, and runtime components
• Supporting internal teams and external stakeholders with technical insights and documen...