AI Software Intern: Shape Next‑Gen AI on Custom Hardware

Tenstorrent; Warszawa; On-site; PLN 36,000 - 51,000; Full time

Job summary

Tenstorrent is seeking a full-time, on-site internship in Warsaw for 13 weeks (6 months preferred) across Training, Models, and Ops/Kernels teams. You will develop and profile distributed ML training workflows, optimize models for our hardware, and build high-performance kernels on Tenstorrent devices.

Your work will span Python/C++ development in a Linux environment, with exposure to ML frameworks like PyTorch, JAX, or TensorFlow.

Qualifications

• Pursuing BS/MS/PhD in CS/CE or related field.
• Comfortable with Python and C++, and Linux development environment.
• Strong fundamentals in data structures, algorithms, and systems.
• Interest in ML workloads on accelerator hardware.

Responsibilities

• Work on distributed ML training workflows and tooling.
• Contribute to model optimization for hardware acceleration.
• Assist kernel and ops optimization for Tenstorrent hardware.
• Collaborate across Training, Models and Ops/Kernels teams.

Skills

Python

C++

Linux

Data structures

Algorithms

Computer architecture

Machine learning workflows

Education

BS/MS/PhD in CS/CE or related

Tools

Git

PyTorch

JAX

TensorFlow

Profiling tools

Job description

Tenstorrent is seeking a full-time, on-site internship in Warsaw for 13 weeks (6 months preferred) across Training, Models, and Ops/Kernels teams. You will develop and profile distributed ML training workflows, optimize models for our hardware, and build high-performance kernels on Tenstorrent devices.

Your work will span Python/C++ development in a Linux environment, with exposure to ML frameworks like PyTorch, JAX, or TensorFlow.