C++ Machine Learning Engineer Models Training

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.

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

Requirements

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