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 Mexico City, Mexico. This role involves developing, supporting, and optimizing the OpenAI/Triton project for GPUs, collaborating with the open-source community, and deploying performance improvements for neural networks implemented with Triton on 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.
- Implementation, documentation, and verification of the project through 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 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 systems.
- GEMM performance fundamentals.
- Experience with Docker.
Languages
- English: B2 Upper Intermediate
Seniority
- Senior
About the Company
[Company description if present]