SR. Software Development Engineer - Triton Kernel Development
Core
Senior engineer optimizing deep learning frameworks (TensorFlow, PyTorch) and GPU kernels for AMD GPUs to accelerate training and inference on multi-GPU and multi-node systems.
Role type
Senior IC software development engineer (GPU kernel optimization)
Builds
Optimized deep learning frameworks and GPU kernels for AMD hardware
Domain
High-performance computing / AI infrastructure
Deliverable
production ML models
Required skills
C++, Python, GPU kernel development, deep learning frameworks, distributed computing, compiler optimization, Linux
Preferred skills
HIP, CUDA, assembly (ASM), AMD architectures (GCN, RDNA), Triton, CUTLASS, Compute Kernel (CK), LLVM, ROCm
Technologies
TensorFlow, PyTorch, HIP, CUDA, Triton, CUTLASS, CK, LLVM, ROCm, Linux
Responsibilities
Optimize deep learning frameworks for AMD GPUs; Develop and optimize GPU kernels for AI operations; Design and optimize deep learning models; Collaborate with internal GPU library teams; Engage with open-source maintainers; Optimize performance on scale-up and scale-out systems; Leverage compiler technologies to improve performance
Seniority
Senior, hands-on IC
