C++ Machine Learning Engineer, AI Models Training

Tenstorrent · Semiconductors · Santa Clara, CA · AI Software

C++ Machine Learning Engineer focused on extending and optimizing the ML training framework for custom silicon, debugging model performance, and supporting production integration.

What you'd actually do

  1. Extend and optimize our ML training framework with new ops, layers, and training features.
  2. Debug and tune model performance on Tenstorrent chips.
  3. Work with compiler and kernel teams to make sure models compile and run as expected.
  4. Support integration of real-world models and help bring them into production.

Skills

Required

  • C++
  • low-level systems programming
  • performance-critical code
  • tensors
  • memory layout
  • compiler graphs
  • PyTorch
  • ML training framework

Nice to have

  • compiler teams
  • kernel teams

What the JD emphasized

  • high-performance C++
  • custom silicon
  • ML training framework
  • production

Other signals

  • custom silicon
  • training framework
  • high-performance C++