C++ Machine Learning Engineer, Models Training
Core
Building high-performance C++ training frameworks and optimizing ML models for custom RISC-V silicon.
Role type
Senior IC machine learning engineer (systems/training)
Builds
High-performance ML training frameworks and optimized model layers for custom AI chips
Domain
AI hardware / Custom silicon / Systems programming
Deliverable
production ML models
Required skills
C++ (low-level, performance-critical), tensor manipulation, memory layout optimization, compiler graph understanding, PyTorch familiarity, kernel development
Preferred skills
RISC-V architecture knowledge, operator fusion techniques, hardware-aware model optimization
Technologies
C++, PyTorch, RISC-V
Responsibilities
Extend and optimize ML training framework with new ops and layers, Debug and tune model performance on Tenstorrent chips, Collaborate with compiler teams to ensure correct model compilation, Support integration of real-world models into production
Seniority
Senior, hands-on IC
