Senior GPU Compiler Software Development Engineer

Job Description

Project Descriptio

n:
Working on GPU support for OpenAI/Triton — a language and compiler for writing highly efficient custom Deep-Learning primitives. Work with the open-source community to analyze, develop, test, and deploy performance improvements for neural networks implemented with Triton on GPUs with RO

Cm.
Responsibilit

ies:
New features development, support and optimization of OpenAI/Triton project for GPUs. Communication with other developers, customers and project managers. Test implementation, project documentation and verification of system with unit/component/functional t

ests.
Mandatory Skills Descri

ption:
• Strong C/C++ programming

skills• Experience with compiler internals (llvm, gcc or any

other)• Basic Python programming

skills• Experience in performance a

nalysis
Nice-to-Have Skills Desc

ription:
• Basic understanding of ML tec

hnologies• Experience with GPGPU (General purpose GPU) computing (HIP, CUDA, Open

CL, etc.)• Experience wit

h PyTorch• Experience with LLVM and MLIR compiler infrastructure, analysis or optimizations imple

mentation• Knowledge of ROCm infra

structure• Experience in CMake, make/ninja bui

ld system• GEMM performance fun

damentals• Experience wi

th Docker

Languages:
English: B2 Upper I