About the Role
As a Senior Machine Learning Engineer on the ML Models team, you will lead the bring-up and optimization of cutting-edge AI models for custom AI devices. You will experiment, optimize, and push boundaries while solving real-world problems, working on models used in practical applications.
Responsibilities
- Bring up and optimize cutting edge AI models on custom AI devices.
- Experiment with and optimize AI models.
- Push boundaries in AI model development.
- Solve real-world problems with AI models.
- Debug performance issues.
- Tune architectures for improved performance.
- Boost model accuracy and robustness.
- Work alongside compiler, kernel, and hardware teams to drive new features, performance optimizations, and fixes.
Requirements
- Hands-on experience bringing up state-of-the-art ML models on new hardware platforms.
- Strong debugging instincts.
- A curiosity-driven mindset that stays current with ML research and brings practical insights to real-world engineering challenges.
- Eligibility to access U.S. export-controlled technology, potentially contingent on citizenship/permanent residency status or ability to obtain prior license approval from the U.S. Commerce Department or applicable federal agency.
Skills
- Python programming
- PyTorch
- Deep learning models
- ML model architectures
- Model optimization techniques (quantization, flash attention, kernel fusing, multi-device parallelization)
- Leading small groups of engineers
- Cross-functional team collaboration
Location
- Toronto, ON
- Remote (on a candidate-by-candidate basis)
Work Type
- Hybrid
- Remote
Experience Level
- Senior
- All seniorities
Salary/Compensations
- $100k - $500k (including base and variable compensation targets)
Benefits
- Highly competitive compensation package
About the Company
- Tenstorrent is leading the industry on cutting-edge AI technology, revolutionizing performance expectations, ease of use, and cost efficiency.
- Solutions must evolve to unify innovations in software models, compilers, platforms, networking, and semiconductors.
- The diverse team has developed a high performance RISC-V CPU from scratch.
- The team shares a passion for AI and a deep desire to build the best AI platform possible.
- Values collaboration, curiosity, and a commitment to solving hard problems.
Equal Opportunity
- Equal opportunity employer.
