Software Engineer, ML Frameworks

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.

We are seeking an experienced and highly skilled Software Engineer with expertise in compilers and semiconductor technology to join our team. As a Software Engineer on the ML Frameworks team, you will play a critical role in designing, developing, and optimizing compilers for cutting-edge semiconductor products. You will work closely with hardware engineers, software engineers, and other stakeholders to ensure the efficient and effective execution of compiler-related tasks.

This role is hybrid, based out of Santa Clara, CA; Austin, TX; or Toronto, ON.

Responsibilities:

• Develop machine learning graph compiler


• Participate in the co-design of Tenstorrent's hardware and software stack


• Benchmark, analyze, and optimize performance of key machine learning applications across Tenstorrent's hardware and software stack


• Develop performance analysis and estimation infrastructure that feeds into Tenstorrent compiler


• Develop high-performance run-time engine


• Integrate the Tenstorrent software into leading machine learning frameworks


• Work closely with machine learning engineers to discover the hardware and software requirements of current and future machine learning applications

Experience & Qualifications:

• BSc, MSc or PhD in Electrical/Computer Engineering or Computer Science


• Experience with algorithms, data structures, and software development in C/C++. Python expertise is welcome as well


• Familiarity with and passion for any of the following -- machine learning, compilers, parallel programming, high-performance and massively parallel systems, processor and computer architecture -- is a plus

Compensation for all engineers at Tenstorrent ranges from $100k - $500k including base and variable compensation targets. Experience, skills, education, background and location all impact the actual offer made.

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.

Our engineering positions and certain engineering support positions require 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/or documentation will be required and considered as Tenstorrent moves through the employment process.

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.