Software Engineer Intern, AI Compiler (Serbia)

Tenstorrent University Jobs • Toronto, ON

76d • Onsite

About The Position

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.
At Tenstorrent, we believe the future of computing must be open, which is why our interns don’t just watch from the sidelines - they help build the core of it. We provide a "code-to-career" pipeline where students collaborate with industry experts to solve high-stakes problems in RISC-V and AI hardware-software co-design. By joining us, you are taking an internship to democratize high-performance computers that are accessible to everyone.
As an intern, you will experience how we do engineering for 3-6 months, working on a carefully defined and real engineering project, guided by a dedicated mentor and a manager, and as a part of a team.
This role is on-site, based in our Belgrade office.

Requirements

• You are a final-year BSc student or higher in Computer Science, Engineering, Math, or a related technical field.
• You have a strong passion for programming and a solid grasp of algorithms and data structures.
• You are proficient in programming languages, particularly Python and C++.
• You are passionate about neural networks and related deep-learning architectures.

Responsibilities

• Design and implement Python/C++ APIs in our SDK for AI/ML developers.
• Extend the model-import pipeline to support new operators, tensor shapes, and graph patterns.
• Develop compiler passes (e.g., rewriting, tiling, optimization) using an MLIR-based infrastructure.
• Design logic for scheduling and executing compiled neural networks efficiently on our custom AI hardware.

What This Job Offers

• Career Level: Intern
• Number of Employees: 101-250 employees