GPU Kernel Engineer for AI/ML Acceleration

Wipro Technologies; Redmond (WA); On-site; USD 60,000 - 149,000; Full time

Job summary

Wipro Limited is seeking an expert-level GPU Software Engineer to support a high-visibility Maya program, building software tooling on top of a custom compiler and SDK. You will develop, optimize, and port GPU kernels and AI workloads to a specialized hardware platform.

The role requires strong Python and C/C++ systems programming, hands-on PyTorch and Triton experience, and a solid understanding of GPU architecture and performance optimization. 5–8 years of experience expected.

Qualifications

• 5-8 years of experience in GPU kernel development.
• Strong Python/C/C++ systems programming.
• Experience with AI workloads and ML frameworks.

Responsibilities

• Develop, optimize, and port GPU kernels.
• Work on AI workloads on specialized hardware platform.
• Collaborate with compiler/SDK teams.

Skills

Python

C/C++

PyTorch

Triton

GPU kernel

GPU architecture

Compiler optimizations

SDKs

Embedded systems

Tools

CUDA

Job description

Wipro Limited is seeking an expert-level GPU Software Engineer to support a high-visibility Maya program, building software tooling on top of a custom compiler and SDK. You will develop, optimize, and port GPU kernels and AI workloads to a specialized hardware platform.

The role requires strong Python and C/C++ systems programming, hands-on PyTorch and Triton experience, and a solid understanding of GPU architecture and performance optimization. 5–8 years of experience expected.