Job Description:

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

Similar jobs

GPU-Kernel Development Engineer

Apply Now
Back to search page