Senior GPU Compiler Software Development Engineer

Project Description:

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 ROCm.

Responsibilities:

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 tests.

Mandatory Skills Description:

• Strong C/C++ programming skills

• Experience with GPGPU (General purpose GPU) computing (HIP, CUDA, OpenCL, etc.) or Experience with compiler internals (llvm, gcc or any other)

• Basic Python programming skills

• Experience in performance analysis

Nice-to-Have Skills Description:

• Basic understanding of ML technologies

• Experience with PyTorch

• Experience with LLVM and MLIR compiler infrastructure, analysis or optimizations implementation

• Knowledge of ROCm infrastructure

• Experience in CMake, make/ninja build system

• GEMM performance fundamentals

• Experience with Docker

Languages:

English: B2 Upper Intermediate

What do we offer our employees?

💰 Tax-Deductible Costs on a contract of employment for all development roles

👩‍⚕️ LuxMed health & dental care, life insurance

🏋🏽‍♀️ MyBenefit program (sports card, well-being program etc.)

🏠 Remote work (or hybrid)

🖋 Stable employment based on an employment contract (or B2B)

🏦 Special offer of banking services

💻 Equipment such as a laptop and monitor

🌎 Support for relocated employees

🎯 Full coverage relocation package for Senior positions

🎓 LuxTalent platform (webinars, training, courses with certificates)

...and even more!