Senior GPU Kernel Developer

Luxoft; Región Centro; Presencial; MXN 400,000 - 600,000; Jornada completa

Descripción de la vacante

A leading technology firm in Mexico is seeking an experienced developer proficient in GPU compute and performance profiling. You will lead efforts to optimize HIP kernels on AMD GPUs, collaborate with teams on GPU-accelerated applications, and ensure performance improvements. The ideal candidate has a strong background in C++, CUDA/HIP programming, and knowledge of AI/ML technologies. This role offers an exciting opportunity to work in a rapidly growing team.

Formación

• Proficiency with C++ and low-level programming (at least C++ 17).
• Proficiency in CUDA or HIP / ROCm 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.
• Experience in AI/ML/DL/NN/NLP/Computer Vision.

Responsabilidades

• Help optimize HIP kernels for specific AMD hardware.
• Collaborate with development teams to optimize and enhance GPU-accelerated applications.
• Debug, profile, and fine-tune code for performance improvements.
• Stay updated with the latest advancements in GPU architectures and programming models.

Conocimientos

C++ proficiency

CUDA or HIP proficiency

GPU architectures knowledge

Parallel programming skills

Python

AI/ML/DL experience

Descripción del empleo

Project description

Luxoft is searching for talented developers with GPU compute and performance profiling experience to join the rapidly growing team. We are seeking an experienced individual proficient in GPGPU applications to join our team. The primary responsibility of this role will be to lead the effort in optimizing HIP kernels on AMD GPUs. The candidate should possess a strong background in GPU computing, parallel programming, and a deep understanding of CUDA or HIP frameworks. Additionally, familiarity with optimization techniques is highly desirable.

Responsibilities

• The main task will be to help optimize HIP kernels for specific AMD hardware
• Collaborate with development teams to optimize and enhance GPU-accelerated applications
• Debug, profile, and fine-tune code for performance improvements
• Stay updated with the latest advancements in GPU architectures and programming models

SKILLS

Must have

• Proficiency with C++ and low-level programming (at least C++ 17)
• Proficiency in CUDA or HIP / ROCm programming
• Solid understanding of GPU architectures, parallel programming models, and optimization techniquesStrong problem-solving skills and the ability to work in a collaborative environment
• One of AI/ML/DL/NN/NLP/Computer Vision experience
• Python

Nice to have

• Linux
• CPU Intrinsics (AVX/SSE)
• GPU Assembler
• Profiling
• gdb/LLDB
• Jinja2 or similar templating engines