We are seeking expertlevel GPU Software Engineers to support a highvisibility platform 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
Key Responsibilities
Develop highperformance GPU kernels for specialized hardware platforms using PyTorch and Triton frameworks
Build software solutions leveraging custom compiler toolchains and SDK capabilities
Design and implement kernellevel optimizations to control and improve hardware execution behavior
Port and optimize opensource AIML models for proprietary or custom SDK environments
Adapt and run highperformance computing benchmarks and stress workloads including
High Performance Linpack HPL
BERT and similar benchmarkstyle workloads
Develop stress testing and validation workloads aligned with hardware behavior and platform reliability goals
Support validation and performance testing for both current and nextgeneration hardware platforms
Collaborate closely with platform architects compiler teams and system engineers to enhance endtoend system performance
Core Technical Skills MustHave
Programming Frameworks
Robust proficiency in Python
Expertise in CC for systemslevel programming
Handson 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
Performance Workloads
Handson experience in
GEMM matrix multiplication kernel development
Porting and optimizing ML models on newcustom hardware platforms
Systemlevel performance tuning benchmarking and stress testing