Job Description
Tenstorrent, a company leading in AI technology, is seeking a Scaleout Intern to join their team. This role involves optimizing low-level workloads, developing kernels, and enhancing software performance for machine learning applications. The intern will collaborate with skilled engineers to ensure software operates efficiently and delivers high-quality results.Responsibilities include:
- Participating in the design, development, and maintenance of kernel-level software components.
- Developing and optimizing kernels and kernel libraries for efficient machine learning and HPC applications.
- Implementing tensor compute and tensor data movement optimizations kernels.
- Analyzing and optimizing low-level code to improve software performance, with a focus on tensor optimization.
- Collaborating with machine learning engineers to integrate optimized kernels into machine learning frameworks.
- Identifying performance bottlenecks and developing strategies to resolve them.
- Writing unit tests and ensuring the stability of kernel-level code.
- Creating documentation for code and APIs.
- Staying updated with the latest developments in kernel development and machine learning.
Requirements:
- Bachelor's degree in Computer Science, Software Engineering, or a related field.
- Experience in kernel development, with a focus on low-level optimizations and tensor optimization.
- Proficiency in C/C++ programming languages.
- Familiarity with machine learning frameworks and concepts.
- Strong problem-solving skills.
- Experience with performance profiling and optimization tools.
- Excellent communication and teamwork skills.
Tenstorrent offers:
- Competitive compensation.
- Opportunity to work on cutting-edge AI technology.