Job Description
Job Role : GPU-Kernel Development Engineer
Primary Skills: GPU, Kernel development, PyTorch, Triton frameworks, AI/ML Model, Python, C++ programming
We are seeking expert-level GPU Software Engineers to support a high-visibility platform initiative within the Maya program, focused on building software tooling on top of a custom compiler and SDK. The role involves developing, optimizing, and porting GPU kernels and AI workloads to a specialized hardware platform.
Core Technical Skills
Programming & Frameworks
- Strong proficiency in Python
- Expertise in C/C++ for systems-level programming
- Hands-on experience with PyTorch
- Experience with Triton (kernel development and Triton language)
GPU & Systems Expertise
- Mandatory experience in GPU kernel development
- Strong understanding of GPU architecture and performance optimization techniques
- Experience with compiler optimizations and runtime execution models
- Familiarity with custom SDKs and hardware abstraction layers
Experience: 3-5 Years .
The expected compensation for this role ranges from $45,000 to $121,000 .
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.
Wipro provides equal employment opportunities to all employees and applicants for employment and prohibits discrimination and harassment of any type without regard to race, color, religion, age, sex, national origin, disability status, genetics, protected veteran status, sexual orientation, gender identity or expression, or any other characteristic protected by federal, state, or local laws. Applications from veterans and people with disabilities are explicitly welcome.
Reinvent your world. We are building a modern Wipro. We are an end-to-end digital transformation partner with the boldest ambitions. To realize them, we need people inspired by reinvention. Of yourself, your career, and your skills. We want to see the constant evolution of our business and our industry. It has always been in our DNA - as the world around us changes, so do we. Join a business powered by purpose and a place that empowers you to design your own reinvention.