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
- Strong 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
- Porting and optimizing ML models on newcustom hardware platforms
- Systemlevel performance tuning benchmarking and stress testing