GPU Engineer

We are looking
for an experienced GPU Programming Engineer to develop, optimize, and deploy
GPU-accelerated solutions for high-performance deep learning workloads. The
role focuses on GPU programming, parallel computing, performance tuning, and
deep learning system integration across platforms such as NVIDIA CUDA, AMD
ROCm/HIP, and OpenCL.

Responsibilities:

• Develop, optimize, and maintain GPU-accelerated
components for deep learning pipelines using CUDA, HIP, or OpenCL.

• Analyze 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 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.

Requirements

Education:

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

• 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.

Need to have (Can be bridged):

• Knowledge and experience in SIMD programming.

• Good understanding of neural network (NN)
operators and hands-on experience with PyTorch, TensorFlow, or TensorRT.

Good to have (Not essential):

• Exposure to deep learning concepts such as
quantization and pruning.

• Experience working with High Performance
Computing (HPC) systems.

Preferred Qualifications (Optional):

Additional
experience in GPU performance engineering, deep learning system optimization,
or heterogeneous computing would be advantageous.