companyTenstorrent logo

C++ Machine Learning Engineer for AI Model Training

TenstorrentSanta Clara, California, United States
Hybrid Full-time

Clicking Apply Now takes you to AutoApply where you can tailor your resume and apply.


Unlock Your Potential

Generate Job-Optimized Resume

One Click And Our AI Optimizes Your Resume to Match The Job Description.

Is Your Resume Optimized For This Role?

Find Out If You're Highlighting The Right Skills And Fix What's Missing

Experience Level

Experience

Qualifications

Who You Are:Proficient in C++ and low-level systems programming, particularly in performance-critical applications. Comfortable working with tensors, memory layout, and compiler graphs. Familiar with PyTorch and interested in how frameworks interact with hardware. A creative problem-solver who delights in tackling technical challenges and exploring the intricacies of system design. What We Need:Enhance and optimize our ML training framework with new operations, layers, and training features. Debug and fine-tune model performance on Tenstorrent silicon. Collaborate with compiler and kernel teams to ensure models compile and execute as intended. Assist in integrating real-world models and facilitate their transition to production. What You Will Learn:How ML frameworks and compilers interconnect at the system level. Strategies for converting training workloads into low-level operations optimized for custom silicon. The fundamentals of large-scale model training, from memory organization to operator fusion. The essentials of building infrastructure that enables rapid iteration in both research and production environments.

About the job

At Tenstorrent, we are at the forefront of pioneering AI technology, setting new benchmarks for performance, usability, and cost-effectiveness. As AI transforms the computing landscape, it is crucial for solutions to evolve, integrating advancements in software models, compilers, platforms, networking, and semiconductors. Our passionate and diverse team of technologists has successfully designed a high-performance RISC-V CPU from the ground up, driven by a shared commitment to AI excellence and the ambition to create the ultimate AI platform. We value collaboration, curiosity, and a relentless pursuit of challenging problems. We are expanding our team and seeking contributors across all experience levels.

As a C++ Machine Learning Engineer on our AI Models team at Tenstorrent, you will play a vital role in developing the training framework for our most sophisticated models. Your responsibilities will involve writing high-performance C++ code, designing new layers and operators, and ensuring our models efficiently scale across our custom silicon. If you have a passion for building the core of machine learning systems and optimizing their performance, this opportunity is tailored for you.

This position is hybrid, based out of Santa Clara, CA.

We encourage candidates with varying levels of experience to apply. During the interview process, we will evaluate candidates to determine the most suitable level, and offers will correspond to that level, which may differ from what is stated in this posting.

About Tenstorrent

Tenstorrent is an innovative leader in AI technology, dedicated to enhancing performance, usability, and cost efficiency. Our team is passionate about developing advanced AI solutions and has achieved remarkable milestones in creating high-performance computing technology.

Similar jobs

Tailoring 0 resumes

We'll move completed jobs to Ready to Apply automatically.