Convolution Operations Development and Optimization Intern (Serbia)

Job Description

Tenstorrent is seeking a Convolution Operations Development and Optimization Intern to join their team in Belgrade, Serbia. The intern will work on cutting-edge AI technology, focusing on convolutions and their optimizations for Tenstorrent hardware. This is an on-site role.

The intern will partner with skilled engineers, explore problem-solving approaches, and work on a defined engineering project. They will also contribute to the development and maintenance of Tenstorrent software components connected to the hardware platform.

Responsibilities:

• Participate in the design, development, and maintenance of specific Tenstorrent software components.
• Develop and optimise kernels for convolution based operations.
• Analyse and optimise low-level code to improve the performance and efficiency of our software, with a strong emphasis on tensor optimisation.
• Collaborate with machine learning engineers and data scientists to integrate optimised kernels and low-level routines into machine learning frameworks and pipelines.
• Identify performance bottlenecks, conduct performance profiling, and develop strategies to address and resolve them.
• Write comprehensive unit tests, conduct thorough debugging, and ensure the stability and reliability of kernel-level code.
• Discuss observations and how you would approach the project assigned to you, as well as how to present your accomplishments/findings to the whole team at the end of your internship.

Requirements:

• Enrolled in the final year of BSc or MSc studies in Computer Science, Computer Engineering, Software Engineering, Electronics, Math, or a related field.
• Familiarity with one or more programming languages, but not limited to: C++ and Python.
• Basic familiarity with machine learning frameworks and concepts.
• Interest in learning about (or expanding knowledge in) kernel development, with a strong focus on low-level optimizations and tensor optimization.
• Interest in performance profiling and optimization tools.

Tenstorrent offers:

• A highly competitive compensation package and benefits.