C++ Machine Learning Engineer, AI Models Training

Tenstorrent is leading the industry on cutting-edge AI technology, revolutionizing performance expectations and cost efficiency. As a C++ Machine Learning Engineer on the AI Models team, you will work on the training framework for advanced models, writing high-performance C++ code and optimizing model performance on custom silicon.

Responsibilities

• Extend and optimize our ML training framework with new ops, layers, and training features
• Debug and tune model performance on Tenstorrent chips
• Work with compiler and kernel teams to make sure models compile and run as expected
• Support integration of real-world models and help bring them into production

Skills

• Strong in C++ and low-level systems programming, especially in performance-critical code
• Comfortable thinking in tensors, memory layout, and compiler graphs
• Familiar with PyTorch and curious about how frameworks map to hardware
• A builder who enjoys solving technical puzzles and digging into the details
• Extend and optimize our ML training framework with new ops, layers, and training features
• Debug and tune model performance on Tenstorrent chips
• Work with compiler and kernel teams to make sure models compile and run as expected
• Support integration of real-world models and help bring them into production

Benefits

• Tenstorrent offers a highly competitive compensation package and benefits
• We are an equal opportunity employer
• This offer of employment is contingent upon the applicant being eligible to access U.S. export-controlled technology

Company Overview

Company H1B Sponsorship