AI Framework Software Development Engineer

TN United Kingdom; Milton Keynes; On-site; GBP 50,000 - 90,000; Full time

Job summary

An innovative company is seeking an AI Framework Software Development Engineer to enhance deep learning applications on AMD GPU platforms. This role involves optimizing frameworks like TensorFlow and PyTorch, collaborating with open-source communities, and applying best software engineering practices. If you are passionate about AI and have strong C/C++ programming skills, this opportunity will allow you to significantly impact the future of computing. Join a team that values innovation, collaboration, and inclusivity, and contribute to transformative technology that enriches lives worldwide.

Qualifications

• Strong skills in C/C++ programming and debugging.
• Experience with large-scale heterogeneous clusters is a plus.

Responsibilities

• Build and optimize deep learning applications for AMD GPU platforms.
• Collaborate with open-source communities for improvements.

Skills

C/C++ Programming

Debugging

Performance Analysis

Deep Learning

Compiler Technologies

GPU Computing

Education

Masters in Computer Science

PhD in Computer Engineering

Tools

TensorFlow

PyTorch

CUDA

OpenCL

Job description

Social network you want to login/join with:

AI Framework Software Development Engineer, Milton Keynes

Client: Advanced Micro Devices, Inc

Location: Milton Keynes, United Kingdom

Job Category: Other

EU work permit required: Yes

Job Reference: 562353d9a422

Job Views: 14

Posted: 01.05.2025

Job Description:

WHAT YOU DO AT AMD CHANGES EVERYTHING. We care deeply about transforming lives with AMD technology to enrich our industry, communities, and the world. Our mission is to build great products that accelerate next-generation computing experiences—fundamentals for data centers, AI, PCs, gaming, and embedded systems. We foster a culture of innovation, collaboration, humility, and inclusivity.

Role: AI Framework Software Development Engineer

Responsibilities:

1. Build and optimize deep learning applications and AI frameworks for AMD GPU platforms.
2. Collaborate with open-source communities to analyze, develop, test, and deploy improvements.
3. Optimize frameworks like TensorFlow, PyTorch on AMD GPUs in upstream repositories.
4. Develop and optimize deep learning models on AMD GPUs.
5. Work with GPU library teams to enhance training and inference processes.
6. Engage with open-source maintainers to integrate code changes upstream.
7. Optimize for multi-GPU and multi-node distributed systems.
8. Work with compiler technologies to improve deep learning pipelines.
9. Apply software engineering best practices.

Preferred Experience:

• Independent work, project scope definition, and leadership skills.
• Strong C/C++ programming, debugging, and performance analysis skills.
• Experience with large-scale heterogeneous clusters is a plus.
• Knowledge of compiler technologies and GPU computing (HIP, CUDA, OpenCL).
• Basic understanding of Deep Learning.

Academic Credentials: Masters or PhD in Computer Science, Computer Engineering, or related fields.

AMD is an equal opportunity employer. We encourage applications from all qualified candidates and will accommodate needs under applicable laws throughout the recruitment process.