AI Software Intern

Tenstorrent University Jobs; Austin, Santa Clara (TX, CA); In loco; USD 69.000 - 96.000; Tempo pieno

Vantaggi offerti da questo lavoro

Competitive compensation

Equal opportunity employer

Descrizione del lavoro

Tenstorrent University Jobs is seeking interns in Austin or Santa Clara to contribute on Kernels, Models, Inference, Scaleout and Runtime teams. You will work on high-performance kernels, model optimization for our hardware, and production-grade ML serving infrastructure.

Candidates are pursuing BS/MS/PhD in CS/CE/Physics/Math with experience in PyTorch, Python and C/C++, and familiarity with CUDA or kernel development. On-site positions in Austin or Santa Clara with competitive compensation.

Competenze

• Pursuing BS/MS/PhD in CS/CE/Physics/Math or related field.

Mansioni

• Develop high-performance kernels on Tenstorrent hardware.
• Optimize ML models for our hardware (LLMs, vision models, video generation).
• Contribute to runtime, memory management, and code execution on hardware.

Conoscenze

C/C++

Python

CUDA

PyTorch

Formazione

BS/MS/PhD in CS/CE/Physics/Math

Strumenti

PyTorch

TensorFlow

JAX

Descrizione del lavoro

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.

Overview

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.

This role ison-site based out of Austin, TX or Santa Clara, CA.

What you might work on

This posting spans multiple teams within Kernels, Models, Inference, Scaleout & our Runtime teams. One application, one recruiter screen, then we match you to the specific team and location that fits best. Teams include:

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

Who You Are

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

What We Need

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

What You Will Learn

• 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

USA Hiring Timelines

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

• Fall Term: Sept-Dec work term, Jan-Aug recruit.

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

Compensation for all interns at Tenstorrent ranges from $50/hr - $70/hr including base and variable compensation targets. Experience, skills, education, background and location all impact the actual offer made.

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.

Voluntary Self‑Identification

For government reporting purposes, we ask candidates to respond to the below self‑identification survey. Completion of the form is entirely voluntary. Whatever your decision, it will not be considered in the hiringprocess or thereafter. Any information that you do provide will be recorded and maintained in aconfidential file.

As set forth in Tenstorrent University Jobs’s Equal Employment Opportunity policy,we do not discriminate on the basis of any protected group status under any applicable law.

If you believe you belong to any of the categories of protected veterans listed below, please indicate by making the appropriate selection.As a government contractor subject to the Vietnam Era Veterans Readjustment Assistance Act (VEVRAA), we request this information in order to measurethe effectiveness of the outreach and positive recruitment efforts we undertake pursuant to VEVRAA. Classification of protected categoriesis as follows:

A "disabled veteran" is one of the following: a veteran of the U.S. military, ground, naval or air service who is entitled to compensation (or who but for the receipt of military retired pay would be entitled to compensation) under laws administered by the Secretary of Veterans Affairs; or a person who was discharged or released from active duty because of a service‑connected disability.

A "recently separated veteran" means any veteran during the three‑year period beginning on the date of such veteran's discharge or release from active duty in the U.S. military, ground, naval, or air service.

An "active duty wartime or campaign badge veteran" means a veteran who served on active duty in the U.S. military, ground, naval or air service during a war, or in a campaign or expedition for which a campaign badge has been authorized under the laws administered by the Department of Defense.

An "Armed forces service medal veteran" means a veteran who, while serving on active duty in the U.S. military, ground, naval or air service, participated in a United States military operation for which an Armed Forces service medal was awarded pursuant to Executive Order 12985.