1 day ago
São Paulo, BrazilSenior
Responsibilities
- Optimize TensorFlow and PyTorch deep learning frameworks for AMD GPUs.
- Design and optimize GPU kernels for AI operations using low-level programming techniques.
- Develop and optimize deep learning models for AMD GPU performance.
- Improve training and inference performance across multi-GPU and multi-node systems.
- Collaborate with internal GPU library teams and open-source maintainers to integrate and upstream optimizations.
- Use compiler technologies and graph compilers to optimize the deep learning pipeline.
- Apply software engineering, debugging, performance tuning, and testing practices to deliver maintainable solutions.
Requirements
- Bachelor’s and/or master’s degree in Computer Science, Computer Engineering, Electrical Engineering, or a related field.
- Experience developing and optimizing GPU kernels for deep learning on AMD GPUs using HIP, CUDA, and assembly.
- Strong knowledge of AMD GCN and RDNA architectures and low-level programming.
- Experience with Compute Kernel, CUTLASS, and Triton for multi-GPU and multi-platform performance.
- Experience integrating optimized GPU performance into TensorFlow or PyTorch for model training and inference.
- Proficiency in Python and C++ with experience in debugging, performance tuning, and test design.
- Experience running large-scale workloads on heterogeneous compute clusters and optimizing scalability.
- Foundational understanding of compiler theory and experience with LLVM and ROCm.
Benefits
- AMD benefits are referenced through its benefits-at-a-glance information, but specific benefits are not provided in the posting.
Categories
About AMD
AMD designs and sells CPUs, GPUs, and adaptive/embedded computing products for PCs, data centers, gaming, and edge devices. Its portfolio includes Ryzen and EPYC processors, Radeon and Instinct graphics, and adaptive SoCs from its Xilinx acquisition, sold to OEMs, cloud providers, and device makers. Founded in 1969 and headquartered in Santa Clara, it is a public company on NASDAQ and supplies semi-custom chips for major game consoles.