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.
Benefits
- Tenstorrent offers a highly competitive compensation package and benefits, and we are an equal opportunity employer.
Work Arrangement
On-site — Belgrade
Additional Information
- This role is on-site, based in our Belgrade office.
- Internship duration: 3-6 months.
- Interns work on a carefully defined and real engineering project.
- Interns are guided by a dedicated mentor and a manager, and work as part of a team.
- Hiring Timelines: 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.
- Actual timelines may vary.
- Offer of employment is contingent upon the applicant being eligible to access U.S. export-controlled technology.
- Due to U.S. export laws, including those codified in the U.S. Export Administration Regulations (EAR), the Company is required to ensure compliance with these laws when transferring technology to nationals of certain countries (such as EAR Country Groups D:1, E1, and E2).
- These requirements apply to persons located in the U.S. and all countries outside the U.S.
- The offer may be contingent upon your citizenship/permanent residency status or ability to obtain prior license approval from the U.S. Commerce Department or applicable federal agency.
- If employment is not possible due to U.S. export laws, any offer of employment will be rescinded.
