ML Software Intern

<div class="content-intro"><p>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.</p></div><div>At Tenstorrent, we believe the future of computing must be open, which is why our interns don’t just watch from the sidelines — they help build the core of it. We provide a code-to-career pipeline where students collaborate with industry experts to solve high-stakes problems in RISC-V and AI hardware-software co-design.  By joining us, you are taking on an internship that helps democratize high-performance computing and make it more accessible to everyone.</div> <div> </div> <div>As a Machine Learning Engineer Intern at Tenstorrent, you will help develop and optimize machine learning algorithms and models on Tenstorrent devices. You will work across software and hardware boundaries to improve performance, robustness, and functionality while collaborating with cross-functional teams on industry-leading AI technologies.</div> <div> </div> <div>This role is a 3-6-month full-time, on-site internship based in our Warsaw, Poland office.</div> <div> </div> <div><br><strong>Who You Are</strong></div> <ul> <li>A student currently pursuing a degree in Computer Science, Computer Engineering, Electrical Engineering, or a related field.</li> <li>Strong in machine learning fundamentals, with particular familiarity in areas such as LLMs and CNNs.</li> <li>Comfortable working with frameworks such as PyTorch and TensorFlow, and programming languages including Python, C++, and CUDA.</li> <li>Familiar with data processing and analysis tools such as NumPy and Pandas, with strong analytical and problem-solving skills.</li> <li>A collaborative communicator who enjoys working across teams and digging into technical details to solve complex problems.</li> </ul> <div><strong>What We Need</strong></div> <ul> <li>Understand how models map onto Tenstorrent devices through compilation flows and kernels, identify gaps in functionality or performance, and propose solutions to address them.</li> <li>Implement new kernels or improve existing kernels for computation and data movement.</li> <li>Design and run experiment