AI Software Intern (Canada)

This is the team that makes "it runs on Tenstorrent hardware" actually true for real models, at real scale, and builds the infrastructure that keeps the rest of engineering moving fast.

Currently pursuing a BS, MS, or PhD in Comp Sci, Comp Eng, Physics/Math or a related field.

Coursework or projects any of the following: parallel processing, machine learning or distributed systems.

A high comfort level as an end user of AI and agentic flows.

Experience with model quantization, kernel fusion, or other optimization techniques; or low level high performance programming.

Hands on experience working with Pytorch, experimenting with and deploying models, for inference or training.

Experience with C/C++ or kernel development in languages like CUDA, or Python and at least one ML framework (PyTorch, TensorFlow, JAX).

Knowledge of RTL, HDL is nice to have on certain teams.

How models get adapted to run well on specialized hardware

How to write high performance code that can get the most out of the underlying hardware

Production ML serving infrastructure at scale

How infrastructure choices affect the velocity of an entire engineering org

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

This 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. As the position offered will have direct and/or indirect access to information, systems, or technologies subject to these laws, 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.