This list contains only the countries for which job offers have been published in the selected language (e.g., in the French version, only job offers written in French are displayed, and in the English version, only those in English).
As a core member of the team, you will play a pivotal role in optimizing and developing deep learning frameworks for AMD GPUs. Your experience will be critical in enhancing GPU kernels, deep learning models, and training/inference performance across multi-GPU and multi-node systems. You will engage with both internal GPU library teams and open-source maintainers to ensure seamless integration of optimizations, utilizing cutting-edge compiler technologies and advanced engineering principles to drive continuous improvement.
Job Responsibility:
End to end optimization: Build and optimize end to end distributed inference and RL solutions on mainstream frameworks like vLLM and SGlang
Collaborate with GPU Library Teams: Work closely with internal teams to analyze and improve training and inference performance on AMD GPUs
Collaborate with Open-Source Maintainers: Engage with framework maintainers to ensure code changes are aligned with requirements and integrated upstream
Work in Distributed Computing Environments: Optimize deep learning performance on both scale-up (multi-GPU) and scale-out (multi-node) systems
Utilize Cutting-Edge Compiler Tech: Leverage advanced compiler technologies to improve deep learning performance
Optimize Deep Learning Pipeline: Enhance the full pipeline, including integrating graph compilers
Software Engineering Best Practices: Apply sound engineering principles to ensure robust, maintainable solutions
Requirements:
Master's or PhD in Computer Science, Computer Engineering, Electrical Engineering, or related fields
5+ years of professional experience in technical software development, with a focus on GPU optimization, performance engineering, and framework development
Nice to have:
GPU Kernel Development & Optimization
Deep Learning Integration
End to end solution optimization
High-Performance Computing
Compiler Optimization
Text to Video or Image to Video experience
What we offer:
Benefits offered are described: AMD benefits at a glance