Senior GPU Compiler Software Development Engineer

Luxoft; Remote; GBP 50,000 - 70,000; Full time

Job summary

A leading technology company in the United Kingdom is seeking a developer to work on GPU support for an OpenAI/Triton project. Responsibilities include new feature development and optimizations, along with communication with various stakeholders. The ideal candidate has strong C/C++ programming skills and experience with compiler internals, among other qualifications.

Qualifications

• Strong C/C++ programming skills required.
• Experience with compiler internals like llvm, gcc preferred.
• Basic Python programming skills a plus.

Responsibilities

• Develop new features and optimize OpenAI/Triton's GPU support.
• Communicate with developers, customers, and project managers.
• Implement tests and document the project.

Skills

Strong C/C++ programming skills

Experience with compiler internals

Basic Python programming skills

Experience in performance analysis

Tools

GPGPU computing

Docker

CMake

Job description

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.

Skills

Must have

• Strong C/C++ programming skills
• Experience with compiler internals (llvm, gcc or any other)
• Basic Python programming skills
• Experience in performance analysis

Nice to have

• Basic understanding of ML technologies
• Experience with GPGPU (General purpose GPU) computing (HIP, CUDA, OpenCL, etc.)
• 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