AI Framework Software Development Engineer
Advanced Micro Devices
- Milton Keynes
- Permanent
- Full-time
- 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 your 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 your knowledge of software engineering best practices
- Ability to work independently, define project goals and scope, and lead your own development effort.
- Excellent C/C++ programming and software design skills, including debugging, performance analysis, and test design.
- Experiences to run workloads on large scale heterogeneous cluster is a plus
- Knowledge of compiler is a plus
- Knowledge of GPU computing (HIP, CUDA, OpenCL) and basic understanding of Deep Learning is a plus
- Masters or PhD or equivalent experience in Computer Science, Computer Engineering, or related field.