Acceleration Kernel Developer Intern

Tenstorrent University Jobs•Toronto, ON

46d•Hybrid

About The Position

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 Acceleration Kernel Developer at Tenstorrent, you will play a crucial role in optimizing low-level workloads, kernel development, and enhancing our software's performance for machine learning applications. You will work closely with a team of highly skilled engineers to ensure that our software operates at peak efficiency, delivering high-quality results to our clients and users. This role is hybrid, based out of Toronto, Ontario.

Requirements

Currently enrolled in a bachelors degree, or a masters or doctorate program in Computer Science, Software Engineering, Computer Engineering, Engineering Science or a related field.
A strong understanding of algorithm implementation and low level algorithm optimization.
Experience with parallel acceleration development, such as parallel algorithms on CPU, or GPU acceleration.
High degree of proficiency in C/C++ and Python programming languages.
Familiarity with machine learning frameworks and concepts.
Strong problem-solving skills and the ability to analyze and debug complex issues.
Experience with debug, performance profiling and optimization tools.
Excellent communication and teamwork skills.
Self-motivated, detail-oriented, and able to work independently as well as in a team.

Nice To Haves

Familiarity with infrastructure built with cloud services is a plus (e.g. AWS).

Responsibilities

Kernel Development: Participate in the design, development, and maintenance of kernel-level software components for our applications. develop and optimize kernels and kernel libraries for efficient machine learning and HPC applications.
Implementation of tensor compute and tensor data movement optimizations kernels
Heavy focus on optimizations.
Low-Level Optimization: Analyze and optimize low-level code to improve the performance and efficiency of our software, with a strong emphasis on tensor optimization.
Machine Learning Integration: Collaborate with machine learning engineers and data scientists to integrate optimized kernels and low-level routines into machine learning frameworks and pipelines.
Performance Profiling: Identify performance bottlenecks, conduct performance profiling, and develop strategies to address and resolve them.
Testing and Debugging: Write comprehensive unit tests, conduct thorough debugging, and ensure the stability and reliability of kernel-level code.
Infrastructure: Develop infrastructure to analyze functional test and performance test data.
Documentation: Create clear and concise documentation for code, APIs, and best practices to facilitate collaboration within the team.
Research and Innovation: Stay up-to-date with the latest developments in kernel development, tensor optimization, and machine learning to propose innovative solutions and improvements.