AI Software Development Eng.
Core
Optimize and develop deep learning frameworks and GPU kernels for AMD GPUs to enhance training and inference performance across multi-GPU and multi-node systems.
Role type
Senior IC software engineer (GPU deep learning frameworks)
Builds
Optimized deep learning frameworks (TensorFlow, PyTorch) and GPU kernels for AMD hardware
Domain
High-performance computing, AI/ML infrastructure, GPU architecture
Deliverable
production ML models
Required skills
C++ development, Linux environments, GPU kernel optimization, deep learning frameworks (TensorFlow, PyTorch), distributed computing, compiler technologies, Python
Preferred skills
HIP, CUDA, assembly (ASM), AMD architectures (GCN, RDNA), Compute Kernel (CK), CUTLASS, Triton, LLVM, ROCm, graph compilers
Technologies
TensorFlow, PyTorch, HIP, CUDA, ROCm, LLVM, CK, CUTLASS, Triton, Linux
Responsibilities
Optimize deep learning frameworks for AMD GPUs, develop and optimize GPU kernels, design and optimize deep learning models, collaborate with GPU library teams, engage with open-source maintainers, optimize performance in distributed computing environments, leverage compiler technologies, enhance the deep learning pipeline
Seniority
Senior, hands-on IC
