Tenstorrent is a leader in cutting-edge AI technology, focusing on revolutionizing performance, ease of use, and cost efficiency. As AI redefines computing, Tenstorrent is unifying innovations in software models, compilers, platforms, networking, and semiconductors. Our diverse team has developed a high-performance RISC-V CPU from scratch and is passionate about building the best AI platform. We value collaboration, curiosity, and solving challenging problems, and we are seeking contributors of all seniorities.
As a Machine Learning Engineer on the AI Models team at Tenstorrent, you will bring up and optimize cutting-edge AI models to run on our custom AI devices. You will experiment, optimize, and push boundaries while solving real-world problems. If you are passionate about the craft of ML and want to work on models used in real-world applications, this role is for you.
This is a hybrid role based in Warsaw or Gdansk, Poland, with consideration for remote candidates on a case-by-case basis. We welcome candidates at various experience levels, and the appropriate level will be assessed during the interview process.
Who You Are
- Proficient in Python programming and experienced with PyTorch for deep learning model development.
- Driven by curiosity and a desire to experiment, with a knack for understanding and improving complex systems.
- Possess a deep understanding of ML model architectures, capable of optimizing individual components and their interactions.
- Collaborative and enjoy working across software and hardware teams.
What We Need
- Experience bringing up state-of-the-art ML models on new hardware platforms.
- Ability to analyze performance bottlenecks, fine-tune architectures, and enhance model accuracy and robustness.
- Familiarity with model optimization techniques (e.g., quantization, sparsity, attention) and key hardware features like matrix engines, caches, and memory hierarchies.
- A strong understanding of current ML research and the ability to translate insights into practical improvements.
What You Will Learn
- How to deploy real ML models on a custom AI accelerator.
- Techniques for optimizing ML model performance from application to silicon level.
- The process of moving from research papers to production-ready ML deployments.
- How to collaborate with compiler, kernel, and hardware teams to drive new features, performance optimizations, and fixes.
Tenstorrent offers a highly competitive compensation package and benefits.
