Convolution Operations Development and Optimization Intern (Serbia)

Tenstorrent is leading the industry on cutting-edge AI technology, revolutionizing performance expectations, ease of use, and cost efficiency. With AI redefining the computing paradigm, solutions must evolve to unify innovations in software models, compilers, platforms, networking, and semiconductors. Our diverse team of technologists have developed a high performance RISC-V CPU from scratch, and share a passion for AI and a deep desire to build the best AI platform possible. We value collaboration, curiosity, and a commitment to solving hard problems. We are growing our team and looking for contributors of all seniorities.

If you wish to immerse yourself in innovative technologies, and you have a passion for programming and solid foundation in algorithms and data structures, you are a great fit for our Tenstorrent internship program! During your 3-month internship, you will partner with highly skilled engineers, experience many approaches to problem-solving and will work on a carefully defined engineering project in relation to convolutions and there optimizations for Tenstorrent hardware, guided by a dedicated mentor.

This role is on-site, based in our Belgrade office.

Responsibilities:

• Software Development: Participate in the design, development, and maintenance of specific Tenstorrent software components (connected to the hardware platform) for our applications. Develop and optimise kernels for convolution based operations.
• Special Program Optimisation: Analyse and optimise low-level code to improve the performance and efficiency of our software, with a strong emphasis on tensor optimisation.
• Machine Learning Integration: Collaborate with machine learning engineers and data scientists to integrate optimised kernels and low-level routines into machine learning frameworks and pipelines.
• Performance Profiling: Identify performance bottlenecks, conduct performance profiling, and develop strategies to address and resolve them.
• Testing and Debugging: Write comprehensive unit tests, conduct thorough debugging, and ensure the stability and reliability of kernel-level code.
• Internship project: Discuss your 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.

Experience & Qualifications:

• 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.
• [Nice to have] Interest in learning about (or expanding knowledge in) kernel development, with a strong focus on low-level optimizations and tensor optimization.
• [Nice to have] Interest in performance profiling and optimization tools.