GPU Software Engineer

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.

This is a critical and time-sensitive engagement with immediate onboarding expectations and long-term roadmap alignment (~18 months)., * Develop GPU kernels for specialized hardware platforms using PyTorch/Triton frameworks

• Build software solutions leveraging custom compiler and SDK capabilities
• Design and implement kernel-level optimizations to control hardware execution behavior
• Port open-source AI/ML models to custom SDK environments
• Port and adapt high-performance computing benchmarks and stress workloads such as:
• Linpack (High Performance Linpack)
• BERT/benchmark-style workloads (referred as “Babu bench”)
• Develop stress testing and validation workloads aligned to hardware behaviour and platform validation
• Support testing and stress testing of current and next-generation hardware platforms
• Collaborate closely with platform architects and compiler teams to enhance system capabilities
• Core Technical Skills (Must-Have)

• Python
• C/C++ (systems-level programming)
• PyTorch
• Triton (Triton language / kernel development)

GPU & Systems Expertise

• GPU kernel development (mandatory and critical)
• Strong understanding of GPU architecture and compute optimization
• Experience with compiler-based optimizations / runtime execution layers
• Experience with custom SDKs or hardware abstraction layers