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-
or continue with email
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-
Sign in
To continue your applicationor continue with email
By continuing you agree to our Terms & Privacy Policy.
Similar jobs
Show more