ML Software Intern

Requirements

• A student currently pursuing a degree in Computer Science, Computer Engineering, Electrical Engineering, or a related field.
• Strong in machine learning fundamentals, with particular familiarity in areas such as LLMs and CNNs.
• Comfortable working with frameworks such as PyTorch and TensorFlow, and programming languages including Python, C++, and CUDA.
• Familiar with data processing and analysis tools such as NumPy and Pandas, with strong analytical and problem-solving skills.
• A collaborative communicator who enjoys working across teams and digging into technical details to solve complex problems.
• What We Need
• Understand how models map onto Tenstorrent devices through compilation flows and kernels, identify gaps in functionality or performance, and propose solutions to address them.
• Implement new kernels or improve existing kernels for computation and data movement.
• Design and run experiments to evaluate the performance of different technical approaches across hardware platforms.
• Improve model and software stack robustness through benchmarking experiments and CI/CD pipeline development.
• Stay current with new research and practical advancements in machine learning and AI, and apply relevant ideas to ongoing projects.
• What You Will Learn
• How machine learning workloads are compiled, executed, and optimized on Tenstorrent hardware.
• How kernel development impacts performance, functionality, and efficiency across the software and hardware stack.
• How to evaluate architectural and implementation tradeoffs using real benchmarking and performance experiments.
• How to collaborate with cross-functional engineering teams to bring machine learning solutions into production environments.
• Tenstorrent offers a highly competitive compensation package and benefits, and we are an equal opportunity employer.

Additional Information

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. 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. 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. This role is a 6-month full-time, on-site internship based in our Warsaw, Poland office.

Your Match

How well this role fits your profile.

Company Intel

What employees say

Worked at tenstorrentuniversity? Share your experience