T

Tenstorrent

Designs AI chips and licenses processor cores

AI Software Engineer Intern

Applied Machine LearningConfirmed live in the last 24 hours
Internship · Year-round$50 - $70 /hr

About the Job

Requirements

  • Currently pursuing a Bachelor of Science, Master of Science, or PhD in Computer Science, Computer Engineering, Physics, Mathematics, or a related field.
  • Coursework or projects in parallel processing, machine learning, or distributed systems.
  • Experience with model quantization, kernel fusion, other optimization techniques, or low-level high-performance programming.
  • Hands-on experience working with PyTorch and experimenting with and deploying models for inference or training.
  • Experience with C/C++, kernel development in languages such as CUDA, or Python and at least one machine learning framework such as PyTorch, TensorFlow, or JAX.

Responsibilities

  • Develop high-performance kernels on Tenstorrent hardware.
  • Optimize machine learning models, including large language models, vision models, and video and image generation models, for Tenstorrent hardware.
  • Develop and optimize inference servers.
  • Build runtime software that manages memory, task scheduling, and code execution on hardware while an application is running.
  • Develop communication and coordination capabilities between devices in a distributed AI system.
  • Create compiler and infrastructure tools that optimize AI models into high-performance programs on Tenstorrent hardware, including memory planning, profiling, debugging, and emulation.

Desired Qualifications

  • Knowledge of register-transfer level and hardware description languages is nice to have for certain teams.

About the Company

Tenstorrent

Designs AI chips and licenses processor cores

Tenstorrent designs AI processors and computing systems for developers, enterprises and chip designers that want control over their hardware. Its Galaxy servers connect multiple AI processors to run language models, video generation and other workloads across larger systems. The company sells accelerator cards, workstations and servers, and licenses RISC-V CPU designs and Tensix AI cores for customers to incorporate into their own chips. Its software compilers translate models from popular machine learning frameworks into programs for Tenstorrent hardware. Founded in 2016, it operates from offices in the United States, Canada, Europe and Asia.

Company size
501-1,000
Company stage
Late Stage VC
Total funding
$2.8B
Headquarters
Santa Clara, CA
Founded
2016

Recently Posted Jobs

Explore more opportunities at Tenstorrent

See more jobs