Tenstorrent is seeking a Software Intern to work on AI compilers, focusing on developing and optimizing machine learning graph compilers and runtime engines. This role involves close collaboration with hardware and software teams to co-design and benchmark performance across Tenstorrent's full stack, based in Austin, TX in a hybrid work model.
What You'll Do
- 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
- Develop novel ML models and primitives that take advantage of Tenstorrent’s breakthrough architecture to deliver orders of magnitude performance & efficiency improvements
What We're Looking For
- Final year BS/MS or PhD candidate in EE/ECE/CE/CS with a strong GPA
- Experience with algorithms, data structures, and software development in C/C++
- Strong problem solving and analytical skills
Nice to Have
- 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
Technical Stack
C/C++, Python, machine learning, compilers, parallel programming, high-performance systems, massively parallel systems, processor architecture, computer architecture
Benefits & Compensation
- Highly competitive compensation package
- Equal opportunity employer
- Opportunity to move between digital design, verification, firmware, software and system engineering teams
- Dedicated mentor for career acceleration
Compensation: $50/hr - $70/hr including base and variable compensation targets
Work Mode
Hybrid work model based in Austin, TX
Tenstorrent offers a highly competitive compensation package and benefits, and we are an equal opportunity employer.
