AI Software Intern: High-Performance ML & Hardware

Tenstorrent University Jobs; Austin, Santa Clara (TX, CA); On-site; USD 69,000 - 96,000; Full time

Benefits offered by this job

Competitive compensation

Equal opportunity employer

Job summary

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.

Qualifications

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

Responsibilities

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

Skills

C/C++

Python

CUDA

PyTorch

Education

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

Tools

PyTorch

TensorFlow

JAX

Job description

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.