Sr. Staff Machine Learning Researcher - Model Training & Optimization
Tenstorrent3 months ago
Toronto, CanadaSenior / Staff+
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.
Categories
AI ResearchML Engineering
About Tenstorrent
Tenstorrent builds AI compute hardware and RISC‑V CPU technology for companies developing training and inference workloads in data centers and at the edge. The business sells accelerator cards and systems with a full software stack, and licenses RISC‑V CPU IP. Founded in 2016 and headquartered in Toronto, it is privately held and led by semiconductor architect Jim Keller.