AI Framework 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 machine-learning framework engineer (GPU)
Builds
Optimized deep learning frameworks (TensorFlow, PyTorch) and GPU kernels for AMD hardware
Domain
AI / High-Performance Computing / GPU Architecture
Deliverable
production ML models
Required skills
C++, Python, GPU kernel development, deep learning frameworks, compiler optimization, distributed computing, Linux, performance tuning
Preferred skills
HIP, CUDA, assembly (ASM), ROCm, LLVM, Triton, CUTLASS, Compute Kernel (CK), GCN, RDNA architectures
Technologies
TensorFlow, PyTorch, HIP, CUDA, ROCm, LLVM, Triton, CUTLASS, Compute Kernel (CK)
Responsibilities
Optimize deep learning frameworks like TensorFlow and PyTorch for AMD GPUs; Develop and optimize GPU kernels for specific AI operations; Design and optimize deep learning models for AMD GPU performance; Collaborate with internal GPU library teams and open-source maintainers; Optimize deep learning performance on scale-up and scale-out systems; Leverage advanced compiler technologies to improve performance; Mentor junior team members
Seniority
Senior, hands-on IC
