GPU Compiler Software Development

Luxoft; Mexico; On-site; PHP 5,474,000 - 7,908,000; Full time

Job summary

Luxoft in the Philippines is seeking a developer to work on GPU support for OpenAI/Triton, contributing to a language and compiler for writing highly efficient deep-learning primitives. You will analyze, develop, test, and deploy performance improvements for Triton on ROCm-enabled GPUs.

The role requires collaboration with open-source communities, other developers, customers and project managers, plus testing and documenting implementations to ensure robust integration with the OpenAI/Triton

Qualifications

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

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 the system with unit/component/functional tests

Skills

Compiler internals

Python programming

Performance analysis

Job 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 the system with unit/component/functional tests

Mandatory Skills Description:

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