GPU Kernel Software Engineer

Wipro Limited (NYSE: WIT, BSE: 507685, NSE: WIPRO) is a leading technology services and consulting company focused on building innovative solutions that address clients’ most complex digital transformation needs. Leveraging our holistic portfolio of capabilities in consulting, design, engineering, and operations, we help clients realize their boldest ambitions and build future‑ready, sustainable businesses. With over 230,000 employees and business partners across 65 countries, we deliver on the promise of helping our customers, colleagues, and communities thrive in an ever‑changing world.

For additional information, visit us at

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

Core Technical Skills

• Programming & Frameworks:
  Python, C/C++, 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

• Develop high‑performance GPU kernels for specialized hardware platforms using PyTorch and Triton frameworks
• Build software solutions leveraging custom compiler tool chains 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

• 5-8 years of experience in embedded software design and GPU kernel development
• Strong proficiency in Python, C/C++ and AI/ML frameworks (PyTorch, Triton)
• Deep knowledge of GPU architecture, performance tuning, compiler optimizations, and system‑level workloads
• Experience porting and optimizing ML models to custom hardware platforms
• Strong analytical and problem‑solving skills

Compensation ranges from $60,000 to $148,500, depending on geography, skills, and experience. Eligible for standard benefits: medical and dental coverage, disability insurance, paid time off (including sick leave), and other paid/unpaid leave options.

Wipro provides equal employment opportunities and prohibits discrimination or harassment of any type without regard to race, color, religion, age, sex, national origin, disability status, genetics, veteran status, sexual orientation, gender identity or expression, or any other characteristic protected by federal, state, or local laws. We welcome applications from veterans and individuals with disabilities. Reasonable accommodations are available throughout the recruitment and selection process.