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
Hands-on experience with PyTorch
Experience with Triton kernel development and Triton language
GPU Systems Expertise
Mandatory experience in GPU kernel development
Solid 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
Hands-on experience in GEMM matrix multiplication kernel development
Porting and optimizing ML models on recent/custom hardware platforms
System-level performance tuning, benchmarking, and stress testing
Disclaimer: This job posting has been aggregated from external source. Role details, content, and availability are subject to change. Applicants are advised to confirm the latest information directly on the company website before applying.