AI Software Engineer
Core
Optimizing and developing deep learning frameworks for AMD GPUs to enhance training and inference performance across multi-GPU and multi-node systems.
Role type
Senior IC AI Software Engineer (GPU Frameworks)
Builds
Deep learning frameworks, GPU kernels, and distributed inference solutions for AMD GPUs
Domain
AI / High-Performance Computing / GPU Architecture
Deliverable
production ML models
Required skills
C++ development, GPU kernel optimization, distributed computing, compiler technology, deep learning frameworks, Linux environments, performance tuning
Preferred skills
HIP, CUDA, assembly (ASM), ROCm, LLVM, vLLM, SGlang, PyTorch, TensorFlow, graph compilers, RL solutions
Technologies
HIP, CUDA, ROCm, LLVM, vLLM, SGlang, PyTorch, TensorFlow, CUTLASS, Triton, Compute Kernel (CK)
Responsibilities
Build and optimize end-to-end distributed inference and RL solutions; Collaborate with GPU library teams to improve training/inference performance; Engage with open-source maintainers for code integration; Optimize deep learning performance on scale-up and scale-out systems; Leverage compiler technologies to improve performance; Enhance the full deep learning pipeline including graph compilers
Seniority
Senior, hands-on IC
