C++ Machine Learning Engineer Models Training

Tenstorrent is an AI-computing company that sells AI hardware and licenses AI and RISC-V intellectual property.

Toronto, Canada
About Tenstorrent

Tenstorrent builds computers for AI, including AI processors, scalable server systems, and open-source software and compiler tooling. It also licenses AI and RISC-V IP for customers building customized silicon.

View jobs by Tenstorrent

Skills

Candidate Availability

Required and preferred rules are kept separate and reflect the wording in the original posting.

About the Role

You will write high-performance C++ code for the ML training framework. You will implement and optimize operators and layers, tune model performance on custom chips, collaborate with compiler and kernel teams, and support production model integration.

Requirements

  • C++
  • Low-level systems programming
  • Performance-critical code optimization
  • Tensor and memory-layout knowledge
  • Compiler graph knowledge
  • PyTorch

Responsibilities

  • Extend and optimize the ML training framework
  • Implement new operators, layers, and training features
  • Debug and tune model performance on Tenstorrent chips
  • Work with compiler and kernel teams
  • Support real-world model integration and production deployment