Deep Learning Open Source Software Engineer (GPU)

AMD

Santa Clara, CA

Full Time

Senior

29 days ago

Job Description

About the Role

AMD is seeking a Deep Learning Open Source Software Engineer (GPU) to work on optimizing deep learning frameworks and models on AMD GPUs. The role involves collaborating with internal teams and open-source maintainers to improve performance and integration of deep learning applications, leveraging cutting-edge hardware and software technologies. The company emphasizes innovation, collaboration, inclusivity, and excellence in execution, aiming to transform lives through technology and address global challenges.

Key Responsibilities

Optimize deep learning frameworks like TensorFlow, PyTorch, etc. on AMD GPUs in upstream open-source repositories
Develop and optimize key Deep Learning models on AMD GPUs
Collaborate and interact with internal GPU library teams to analyze and optimize training and inference for deep learning
Work with open-source framework maintainers to understand their requirements and have code changes integrated upstream
Work in a distributed computing setting to optimize for both scale-up (multi-GPU) and scale-out (multi-node) systems
Work with cutting-edge compiler technologies
Optimize the entire deep learning pipeline including graph compiler integration
Apply knowledge of software engineering best practices

Requirements

Ability to work independently, define project goals and scope, and lead development efforts
Excellent Python and/or C++ programming and software design skills, including debugging, performance analysis, and test design
Academic credentials: Masters or PhD or equivalent experience in Computer Science, Computer Engineering, or related field

Nice to Have

Experience to run workloads on large scale heterogeneous clusters
Experience with SGLang development and optimization
Knowledge of compiler technology
Knowledge of GPU computing (HIP, CUDA) and basic understanding of Deep Learning

Qualifications

Masters or PhD or equivalent experience in Computer Science, Computer Engineering, or related field

Benefits & Perks

Competitive base pay and total rewards package
Eligibility for incentives such as annual bonuses or sales incentives
Opportunity to own shares of AMD stock and participate in Employee Stock Purchase Plan
Access to competitive benefits (details provided separately)

Working at AMD

AMD emphasizes innovation, collaboration, humility, inclusivity, and execution excellence. The company values diverse perspectives and strives to transform lives through technology, fostering a culture of pushing technological limits to solve important challenges.

Apply Now