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.
#J-18808-Ljbffr