**Project description**:
**Responsibilities**:
Develop ML kernels for matrix multiplication, Flash Attention and other ML operators
Benchmark, perform competitive analysis and optimize kernels to improve performance
Collaborate with the GPU architecture team to improve future generations
**Skills**:
Must have
Proficiency with C/C++
Proficiency in CUDA or HIP / ROCm or OpenCL programming
Solid understanding of GPU architectures, parallel programming models, and optimization techniques
Strong problem-solving skills and the ability to work in a collaborative environment
Nice to have
Familiarity with models like LLama, Mixtral and Gemma is a plus
Knowledge of MLIR, LLVM and GPU assembly and GPU architecture is a plus
Familiarity with PyTorch or JAX
**Other**:
Languages
English: B2 Upper Intermediate
Seniority
Senior
Remote Mexico, Mexico
Req. VR-
C/C++
Automotive Industry
30/01/2025
Req. VR-

📌 Senior Kernel Developer (México)
🏢 Luxoft
📍 México