AI/ML SDK Software Engineering 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.

As an intern at Tenstorrent, you will experience how we do engineering for 3+ months, working on a carefully defined and real engineering project, guided by a dedicated mentor and a manager, and as a part of a team.

This role is on-site, based in our Belgrade office.

Responsibilities

• Implement, in C++ and Python, neural network graph transformations and logic that schedules and executes neural network operations on our AI/ML processors and systems. Optimize for high performance, high resource utilization, low latency, and low power consumption.
• Develop tools to analyze and visualize performance, hardware utilization, placement, routing, and power consumption.
• Implement new APIs in our SDK using Python and C++, to meet the latest needs of AI and machine learning application developers, as our customers.
• Implement data processing operations in C language, as runtime firmware for the AI-specialized engines. The operations are building blocks for scheduling whole neural networks on our chips and systems.
• Implement the data movement operations as C++ software that generates runtime firmware code for the AI-specialized engines, their RISC-V cores and interconnect units.
• Optimize and run modern neural networks on our chips and systems.

Qualifications

• At least final year of BSc studies in Computer Science, Computer Engineering, Software Engineering, Electronics, Math, or a related field.
• Passion for programming and solid foundation in algorithms and data structures.
• Passion for neural networks and related deep-learning architectures.
• Experience and proficiency in one or more programming languages, but not limited to: Python and C++.

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

Due to U.S. Export Control laws and regulations, Tenstorrent is required to ensure compliance with licensing regulations when transferring technology to nationals of certain countries that have been licensing conditions set  by the U.S. government.

As this position will have direct and/or indirect access to information, systems, or technologies that are subject to U.S. Export Control laws and regulations, please note that citizenship/permanent residency, asylee and refugee information and supporting documentation will be required and considered as a condition of employment.

If a U.S. export license is required, employment will not begin until a license with acceptable conditions is granted by the U.S. government.  If a U.S. export license with acceptable conditions is not granted by the U.S. government, then the offer of employment will be rescinded.