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