Senior GPU Compiler Software Development Engineer

Luxoft; Ciudad de México; Presencial; MXN 738,000 - 1,108,000; Jornada completa

Descripción de la vacante

A technology consulting company in Mexico City is looking for a skilled developer to work on GPU support for the OpenAI/Triton project. Responsibilities include developing new features, supporting existing implementations, and optimizing performance. The ideal candidate will have strong C/C++ programming skills and experience with compiler internals. This role offers an opportunity to engage with the open-source community and contribute to cutting-edge technology in deep learning.

Formación

• Strong understanding of programming languages, specifically C/C++.
• Familiarity with compiler design and optimization techniques.
• Basic knowledge of Python for scripting and automation.

Responsabilidades

• Develop new features and optimize existing ones in the OpenAI/Triton project.
• Communicate with developers, customers, and project managers.
• Test implementation and verify system functionality.

Conocimientos

Strong C/C++ programming skills

Experience with compiler internals

Basic Python programming skills

Experience in performance analysis

Herramientas

Docker

CMake

GPGPU computing (HIP, CUDA, OpenCL)

LLVM

Descripción del empleo

Overview

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