Senior Machine Learning Engineer, ML Models

Optimize ML models for custom AI devices.

• •Lead bringing up and optimizing state-of-the-art ML models to run on Tenstorrent's custom AI devices, working across hardware, compiler, and kernel teams to deploy production-ready models.
• •Key Responsibilities Bring up ML models on new hardware platforms and optimize performance.
• •Investigate and debug performance issues, tune architectures, and improve robustness.
• •Apply model optimization techniques like quantization and kernel fusing.
• •Collaborate with compiler, kernel, and hardware teams to drive features and fixes.
• •Requirements Strong Python programming skills and hands-on PyTorch experience.
• •Experience with deploying ML models on custom accelerators or new hardware.
• •Knowledge of model optimization techniques (quantization, attention optimizations, parallelization).
• •Ability to lead small teams and work cross-functionally.