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 our most advanced models, writing high-performance C++ code and optimizing model performance.
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
- Highly competitive compensation package
- Benefits
Company Overview
- Tenstorrent develops AI hardware and software solutions for data processing and machine learning application. It was founded in 2016, and is headquartered in Toronto, Ontario, CAN, with a workforce of 501-1000 employees. Its website is http://tenstorrent.com.