C++ Machine Learning Engineer, Models Training

Extend and optimize the ML training framework by implementing new ops, layers, and training features with a focus on performance.
Debug and tune model performance on our custom silicon, working closely with hardware teams.
Collaborate with compiler and kernel teams to ensure models compile and run as expected.
Support integration of real-world models and help bring them into production.
Work on memory layout and tensor concepts to map frameworks efficiently to hardware for scalable training.
Learn how ML frameworks connect to systems, translating workloads into low-level operations optimized for fast iteration.