Job Description, Responsibilities & Requirements
About the Position
We are seeking a Senior GPU Compiler Software Development Engineer to work on GPU support for OpenAI/Triton in Guadalajara, Mexico. This role involves developing, optimizing, and supporting highly efficient custom Deep-Learning primitives for GPUs using ROCm.
Responsibilities
- Development of new features and optimization of the OpenAI/Triton project for GPUs.
- Communication with other developers, customers, and project managers.
- Testing implementation, project documentation, and verification of the system with unit/component/functional tests.
Requirements
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.
Languages
- English: B2 Upper Intermediate
Seniority
- Senior
About the Company
[Company description if present]