Role description Role Description 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