Acceleration Kernel Developer

Job Description

Tenstorrent is seeking an Acceleration Kernel Developer to contribute to cutting-edge AI technology. The candidate will play a crucial role in optimizing low-level workloads, developing kernels, and enhancing software performance for machine learning applications. This is a hybrid role based in Warsaw or Gdansk, Poland, and involves working closely with a team of skilled engineers to ensure peak software efficiency.Role involves:

• Designing, developing, and maintaining kernel-level software components.
• Developing and optimizing kernels and kernel libraries for machine learning and HPC applications.
• Implementing tensor compute and tensor data movement optimizations kernels.
• Analyzing and optimizing low-level code, with a focus on tensor optimization.
• Collaborating with machine learning engineers to integrate optimized kernels into frameworks.
• Identifying performance bottlenecks and developing strategies to resolve them.
• Writing unit tests, debugging, and ensuring the stability of kernel-level code.
• Creating documentation for code, APIs, and best practices.
• Staying updated with the latest developments in kernel development and machine learning.

Requirements:

• Bachelor’s degree in Computer Science, Computer Engineering, or a related field.
• Proven 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 and the ability to analyze and debug complex issues.
• Experience with 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.
• Comfortable working in a Linux environment.

Tenstorrent offers:

• A highly competitive compensation package and benefits.