Software Engineer Intern, AI Compiler (Serbia)

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.

• 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.

• 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.

• Gain practical experience in ML compiler development and MLIR infrastructure.
• Master high-performance programming in C++ and professional API design in Python.
• Understand hardware-software co-design principles for low-latency AI execution.
• Develop tools for performance analysis and visualization (e.g., hardware utilization, power consumption).

Hiring Timelines

This internship opportunity is available throughout our 3 terms with the following corresponding recruitment cycles:

• Winter Term: Mar–May work term, Nov–Jan recruit.
• Summer Term: Aug–Oct work term, Jan–May recruit.
• Fall Term: Oct–Dec work term, Apr–May recruit.

Please note these timelines are for reference only. Actual timelines may vary.

Tenstorrent offers a highly competitive compensation package and benefits, and we are an equal opportunity employer.