GPU Software Engineer

Position: GPU Software Engineer

Remote

Fill time

Role Summary

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

Key Responsibilities

• 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)

Programming & Frameworks
• 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