AI Software Intern

Tenstorrent develops high-performance AI computing technology spanning software, compilers, platforms, networking, and semiconductors. The AI Software Intern will contribute to teams developing kernels, optimizing machine learning models, building inference and runtime systems, scaling distributed AI systems, and creating compiler and infrastructure tools for Tenstorrent hardware.

What you'll do:

• Kernels: Develops high performance kernels on Tenstorrent hardware
• Models: Optimizes ML models (LLMs, vision models, video and image generation, and other architectures) for our hardware
• Inference Server: development and serving-side optimization
• Runtime: Builds the software engine that manages memory, task scheduling, and code execution on hardware while an application is actively running
• Scale Out: communication and coordination between devices in a distributed AI system
• Compiler & Infra: Create tools that optimize AI models into high-performance programs on Tenstorrent hardware, covering memory planning, profiling, debug, and emulation

What they're looking for:

• 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)
• 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
• Knowledge of RTL, HDL is nice to have on certain teams