GPU Engineer

We are seeking an experienced GPU Programming Engineer to join our team. In this role, you will focus on developing, optimising, and deploying GPU-accelerated solutions for highperformance deep learning workloads. The ideal candidate has strong expertise in GPU programming across one or more platforms. (e.g., NVIDIA CUDA or AMD ROCm/HIP, or OpenCL) and is comfortable working at the intersection of parallel computing, performance tuning, and DL system integration

Location : Bengaluru

Key Responsibilities :
• Develop, optimize, and maintain GPU-accelerated components for deep learning pipelines using frameworks such as CUDA, HIP, or OpenCL

• Analyse and improve GPU kernel performance through profiling, benchmarking, and resource optimization.

• Optimize memory access, compute, throughput, and kernel execution to improve overall system performance on the target GPUs.

• Port existing CPU-based implementations to GPU platforms while ensuring correctness and performance scalability.

• Work closely with system architects, software engineers, and domain experts to integrate GPU-accelerated solutions.

Required Qualifications :

• Bachelor's or master's degree in computer science, Electrical Engineering, or a related field.

• 2+ years of hands-on experience in GPU programming, preferably using CUDA, or other GPU APIs like HIP, OpenCL etc.,

• Strong understanding of GPU architecture, memory hierarchy, shared memory, bank conflicts and parallel programming models.

• Proficiency in C/C++ and hands-on experience developing on Linux-based systems.

• Familiarity with profiling and tuning tools such as Nsight, rocprof, or Perfetto

Good to have skills in addition to GPU :
• Knowledge and Experience in SIMD Programming

• Good understanding of NN Operators & Hands-on experience with PT, TF, Tensor RT.

• Exposure to DL Concepts like Quantization, Pruning etc., • Experience in working with High Performance Compute (HPC) Systems