Job Title: GPU Kernel Software Engineer

City: Redmond

State/Province: Washington

Posting Start Date: 5/14/26

Job Description:

Job Role : GPU Kernel Software Engineer

Primary Skills: GPU Kernel, Pytorch & Triton framework, SDK, AI/ML Model, Python programming, C/C++ system level programming

Core Technical Skills:

Programming & Frameworks: Python Programming, C/C++ for system level programming, PyTorch, Triton (Kernel development)

GPU & Systems Expertise: GPU kernel development, GPU architecture and performance optimization techniques, compiler optimizations and runtime execution models, custom SDKs and hardware abstraction layers

Performance & Workloads: GEMM (matrix multiplication) kernel development, Porting and optimizing ML models on new/custom hardware platforms, System-level performance tuning, benchmarking, and stress testing

Key Responsibilities:͏

• Develop high-performance GPU kernels for specialized hardware platforms using PyTorch and Triton frameworks
• Build software solutions leveraging custom compiler toolchains and SDK capabilities
• Design and implement kernel-level optimizations to control and improve hardware execution behavior
• Port and optimize open-source AI/ML models for proprietary or custom SDK environments
• Adapt and run high-performance computing benchmarks and stress workloads, including:

• High Performance Linpack (HPL)
• BERT and similar benchmark-style workloads

• Develop stress testing and validation workloads aligned with hardware behavior and platform reliability goals
• Support validation and performance testing for both current and next-generation hardware platforms
• Collaborate closely with platform architects, compiler teams, and system engineers to enhance end-to-end system performance

Mandatory Skills: Embedded Software Design.

Experience: 5-8 Years.

The expected compensation for this role ranges from $60,000 to $148,500 .

Final compensation will depend on various factors, including your geographical location, minimum wage obligations, skills, and relevant experience. Based on the position, the role is also eligible for Wipro's standard benefits including a full range of medical and dental benefits options, disability insurance, paid time off (inclusive of sick leave), other paid and unpaid leave options.

Applicants are advised that employment in some roles may be conditioned on successful completion of a post-offer drug screening, subject to applicable state law.