Sr. Staff Machine Learning Researcher - Model Training & Optimization
Tenstorrent1 day ago
Toronto, CanadaSenior / Staff+
H1B Sponsor
Base Salary
$100k - $500k/yr
Responsibilities
- Lead research and development efforts focused on LLM training and inference optimization.
- Train, evaluate, and optimize state-of-the-art AI models on Tenstorrent hardware.
- Improve performance through techniques such as speculative decoding, quantization, kernel fusion, flash attention, and distributed training.
- Investigate system bottlenecks and collaborate cross-functionally to drive performance improvements.
- Translate cutting-edge ML research into scalable, production-ready solutions.
Requirements
- Strong Python and PyTorch experience developing and training deep learning models.
- Deep understanding of ML architectures, LLM training, and inference optimization.
- Hands-on experience training large-scale machine learning models.
- 4+ years of industry and/or academic experience in ML research and LLM development.
- PhD, published research, or experience with speculative decoding is highly valued.