Sr. Staff Machine Learning Researcher - Model Training & Optimization
Core
Leading research and development of large-scale machine learning models, specifically LLM training and inference optimization, on custom AI accelerators.
Role type
Sr. Staff Machine Learning Researcher (Model Training & Optimization)
Builds
Production-ready AI systems and optimized inference performance on Tenstorrent hardware.
Domain
AI/ML research and high-performance computing hardware
Deliverable
production ML models
Required skills
Python, PyTorch, LLM training, inference optimization, speculative decoding, quantization, kernel fusion, flash attention, distributed training
Preferred skills
PhD, published research
Technologies
PyTorch, Tenstorrent hardware
Responsibilities
Lead R&D efforts for LLM training and inference optimization; Train, evaluate, and optimize state-of-the-art AI models; Improve performance via techniques like speculative decoding and quantization; Investigate system bottlenecks; Translate cutting-edge ML research into scalable solutions.
