Senior GPU Compiler Software Development Engineer

Luxoft; Región Centro; Presencial; MXN 738,000 - 1,108,000; Jornada completa

Descripción de la vacante

A global software consulting firm is looking for a developer to work on GPU support for the OpenAI/Triton project. Ideal candidates should have strong C/C++ programming skills, experience with compiler internals, and a basic understanding of Python. Responsibilities include developing new features, supporting optimization efforts, and communicating with other team members. Experience with GPGPU computing and ML technologies is a plus.

Formación

• Experience with GPGPU computing technologies such as HIP, CUDA, OpenCL is preferred.
• Basic understanding of ML technologies is a plus.
• Experience with PyTorch and LLVM or MLIR compiler infrastructure is beneficial.

Responsabilidades

• Develop new features and optimization for the OpenAI/Triton project.
• Communicate with developers and project managers for project requirements.
• Implement tests and ensure project documentation is accurate.

Conocimientos

Strong C/C++ programming skills

Experience with compiler internals (llvm, gcc, or any other)

Basic Python programming skills

Experience in performance analysis

Herramientas

Docker

CMake

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