Sr. Staff Machine Learning Researcher - Model Training & Optimization

Tenstorrent$100K — $500K *
Information Technology
Less than 5 years of experience
Job Overview by Ladders

Qualifications

  • Strong Python and PyTorch skills for developing and training deep learning models.
  • Deep understanding of machine learning architectures, large language model (LLM) training, and inference optimization.
  • Hands-on experience with training large-scale machine learning models.
  • A minimum of 4 years in industry or academia in ML research and LLM development.
  • PhD or published research experience is highly valued, particularly relating to speculative decoding.

Responsibilities

  • Lead research and development focusing on LLM training and inference optimization.
  • Train, evaluate, and optimize AI models specifically on Tenstorrent hardware.
  • Enhance model performance using techniques like speculative decoding and distributed training.
  • Analyze system bottlenecks and work with other teams to improve performance.
  • Translate advanced ML research into production-ready solutions.

Benefits

  • Competitive compensation package with variable targets.
  • Flexible hybrid work environment based in Toronto.
  • Opportunities to work on cutting-edge AI technologies and hardware.
  • Collaborative environment focusing on cross-functional teamwork in ML and hardware optimization.
Full Job Description
Tenstorrent is building next-generation AI systems that push the boundaries of model training, inference, and large-scale distributed compute. The ML Models team sits at the intersection of cutting-edge AI research and high-performance hardware, bringing state-of-the-art machine learning models to life on Tenstorrent's custom AI accelerators. From training large language models to optimizing inference performance at scale, this team works across the full stack to turn breakthrough research into production-ready AI systems. If you are passionate about model training, ML systems optimization, and advancing the frontier of AI research, this is an opportunity to shape how future AI models are developed and deployed.

This role is hybrid, based out of Toronto, ON.

We welcome candidates at various experience levels for this role. During the interview process, candidates will be assessed for the appropriate level, and offers will align with that level, which may differ from the one in this posting.

Who You Are
  • 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.

What We Need
  • 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.

What You Will Learn
  • How to optimize AI models on custom AI accelerators from application to silicon.
  • How large-scale ML systems are deployed, tuned, and scaled in production.
  • How hardware, compiler, kernel, and ML teams collaborate to maximize performance.
  • The challenges and tradeoffs of scaling modern AI workloads across custom hardware.


Compensation for all engineers at Tenstorrent ranges from $100k - $500k including base and variable compensation targets. Experience, skills, education, background and location all impact the actual offer made.

Tenstorrent offers a highly competitive compensation package and benefits.

About Tenstorrent

Tenstorrent is a semiconductor company that designs and develops computer processors for artificial intelligence and machine learning applications. The company's processors are designed to be energy-efficient and scalable, and are used by a range of businesses, from small startups to large enterprises. Tenstorrent was founded in 2016 and is headquartered in Toronto, Ontario.
Learn more about Tenstorrent
Size
51 employees
Industry
Founded
2016

Similar Jobs

More Jobs at Tenstorrent

More Information Technology Jobs

Find similar Sr. Staff Machine Learning Researcher - Model Training & Optimization jobs: