Deep Learning Compiler Development (DSE) Intern (f/m/d)

Location: Cologne, North Rhine-Westphalia, DE

Company: Advanced Micro Devices

Apply now

Apply for Job


What you do at AMD changes everything 


At AMD, we push the boundaries of what is possible.  We believe in changing the world for the better by driving innovation in high-performance computing, graphics, and visualization technologies – building blocks for gaming, immersive platforms, and the data center. 


Developing great technology takes more than talent: it takes amazing people who understand collaboration, respect, and who will go the “extra mile” to achieve unthinkable results.  It takes people who have the passion and desire to disrupt the status quo, push boundaries, deliver innovation, and change the world.   If you have this type of passion, we invite you to take a look at the opportunities available to come join our team.

 

At AMD, we push the boundaries of what is possible.  We believe in changing the world for the better by driving innovation in high-performance computing, graphics, and visualization technologies – building blocks for gaming, immersive platforms, and the data center. 


Developing great technology takes more than talent: it takes amazing people who understand collaboration, respect, and who will go the “extra mile” to achieve unthinkable results.  It takes people who have the passion and desire to disrupt the status quo, push boundaries, deliver innovation, and change the world.   If you have this type of passion, we invite you to take a look at the opportunities available to come join our team.

 

We’re looking for a Deep Learning Compiler Development Intern (f/m/d) to help us build the compiler and software tool chain for deploying custom ML networks optimized for latency, throughput and power to ACAP, for a duration of 6 months at our Cologne office starting as soon as possible. ACAP is a unique platform in the industry that allows to heavily adapt and re-configure the hardware to target any custom ML network, thereby providing best power-performance at the fastest time to market. Your work in the compiler team will enable that!
 

Your tasks:

  • 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 work on hardware performance modeling tools that predict the performance with various level of accuracy and abstraction to be used as part of the compiler optimizations/explorations.
    You will formulate, code and test optimization algorithms and constraints.

 

Your profile:

  • You’re studying towards a Master or PhD in Computer Science, Electrical Engineering or a related field
  • You have strong programming skills (preferably C++ and Python)
  • You have been exposed to AI accelerator architectures
  • 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


We offer:

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



Requisition Number: 176164 
Country/Region/Location: Germany State/Province: North Rhine-Westphalia City: Cologne 
Job Function: 
Student/ Intern/ Temp  

Apply now

Apply for Job

Share this Job