ML Software Intern

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.

• 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.

• 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.

• 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.

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.