Lead Engineer, ML Models

Tenstorrent is leading the industry on cutting-edge AI technology, revolutionizing performance expectations, ease of use, and cost efficiency. With AI redefining the computing paradigm, solutions must evolve to unify innovations in software models, compilers, platforms, networking, and semiconductors. Our diverse team of technologists have developed a high performance RISC-V CPU from scratch, and share a passion for AI and a deep desire to build the best AI platform possible. We value collaboration, curiosity, and a commitment to solving hard problems. We are growing our team and looking for contributors of all seniorities.

As a Lead Machine Learning Engineer on the AI Models team at Tenstorrent, you'll guide the development and optimization of cutting-edge AI models for our custom AI devices. You'll take the lead on shaping technical direction and ensuring the team delivers impactful results. If you enjoy blending hands-on ML engineering with mentoring and strategic decision-making, you'll thrive here.

This role is remote, based in Cyprus.

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

• Confident with Python programming and hands-on experience with PyTorch for developing deep learning models.
• Possess a deep understanding of ML model architectures, with the ability to optimize both individual components and overall model performance.
• Skilled at mentoring engineers, providing technical guidance, and fostering a collaborative team culture.
• Comfortable leading cross-functional discussions, setting priorities, and aligning efforts toward impactful results.

What We Need

• Lead the bring-up of state-of-the-art ML models on new hardware platforms.
• Direct debugging and performance tuning efforts to improve accuracy, efficiency, and robustness.
• Oversee model optimization techniques, like quantization, flash attention, kernel fusing, and multi-device parallelization.
• Champion a curiosity-driven approach by staying ahead of ML research trends and guiding the team in applying relevant advances to solve real-world engineering challenges.

What You Will Learn

• How to lead the deployment of real ML models at scale on a custom AI accelerator.
• Methods to coordinate model optimization from application to silicon level.
• Approaches for bridging research innovations with production-ready engineering.
• How to influence compiler, kernel, and hardware roadmaps through model performance insights.

Tenstorrent offers a highly competitive compensation package and benefits, and we are an equal opportunity employer.

This offer of employment is contingent upon the applicant being eligible to access U.S. export-controlled technology.  Due to U.S. export laws, including those codified in the U.S. Export Administration Regulations (EAR), the Company is required to ensure compliance with these laws when transferring technology to nationals of certain countries (such as EAR Country Groups D:1, E1, and E2).   These requirements apply to persons located in the U.S. and all countries outside the U.S.  As the position offered will have direct and/or indirect access to information, systems, or technologies subject to these laws, the offer may be contingent upon your citizenship/permanent residency status or ability to obtain prior license approval from the U.S. Commerce Department or applicable federal agency.  If employment is not possible due to U.S. export laws, any offer of employment will be rescinded.