C++ Machine Learning Engineer, AI Models… at Tenstorrent · Santa…

Summary: Tenstorrent is leading the industry on cutting-edge AI technology, revolutionizing performance expectations, ease of use, and cost efficiency. As a C++ Machine Learning Engineer on the AI Models team, you’ll work on the training framework behind 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 Required Qualifications: - 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 Required Skills: C++, Low-level systems programming, Performance optimization, Tensor operations, Memory layout, Compiler graphs, PyTorch, Machine learning training frameworks, Model performance tuning, Compiler integration, Kernel development, Custom silicon optimization, Curiosity Benefits: Highly competitive compensation package, Equal opportunity employer