Deep Learning Compiler Development (MLIR) Intern

Key Skills

AI technology, Analytical and Problem solving, Apache TVM, C++, Deep Learning, Effective communication skills, PyTorch, TensorFlow

Job Description

WHAT YOU DO AT AMD CHANGES EVERYTHING

We care deeply about transforming lives with AMD technology to enrich our industry, our communities, and the world. Our mission is to build great products that accelerate next-generation computing experiences – the building blocks for the data center, artificial intelligence, PCs, gaming and embedded. Underpinning our mission is the AMD culture. We push the limits of innovation to solve the world’s most important challenges. We strive for execution excellence while being direct, humble, collaborative, and inclusive of diverse perspectives. This is who we are at our best. One Company. One Team. 

AMD together we advance_

LOCATION: Cologne (DE)

ONSITE: This role requires the student to work from the Cologne office throughout the duration of the internship 

PROGRAM TERM:

• Fall: 1st September 2023 - duration 3 to 6 months 

WHAT YOU’LL BE DOING:

• You will contribute, develop and enhance our ML compiler by building custom optimizations and transformations targeted to ACAP on top of open-source technology like LLVM, TVM, MLIR.
• You will code and test compiler passes.
• You will help to enhance our compiler build, test and development infrastructure.

WHO WE’RE LOOKING FOR:

• You’re studying towards a Master or PhD in Computer Science, Electrical Engineering or a related field
• You have strong knowledge of compiler design
• You have strong C++ programming skills
• You have gained first experience with either using AI accelerators or compiling to an accelerator/ASIC
• You are familiar with deep learning frameworks (e.g. Tensorflow, Pytorch) and software stack (e.g. TVM)
• You have strong communication and problem-solving skills, have a hands-on mentality and are able to work in a team
• You are an EU citizen or possess a valid work permit. Visa sponsorship is not available for this position

BENEFITS:

• An opportunity to work with a breakthrough product shaping a new era of computing
• Professional guidance from our experiences engineerin team with your own mentor
• Full responsibility for your own tasks and projects, and the opportunity to grow personally and professionally
• A modern workplace in Cologne that celebrates the diversity and well-being of our team