Director/Tech Lead, Acceleration Kernel Developer

Job Description

Tenstorrent is seeking a Director/Tech Lead, Acceleration Kernel Developer to guide the optimization of low-level workloads and enhance software performance for machine learning applications. The candidate will lead a team of skilled engineers, ensuring software operates efficiently and delivers high-quality results. This is a hands-on technical role requiring contributions to software development and team leadership. The role is hybrid based in Warsaw or Gdansk, Poland, with fully remote options within Poland considered.Responsibilities:

• Lead the design, development, and maintenance of acceleration kernel software components.
• Develop and optimize kernels and kernel libraries for efficient machine learning and HPC applications.
• Lead the implementation and optimization of tensor compute and tensor data movement kernels.
• Drive efforts in analyzing and optimizing low-level code, with a strong emphasis on tensor optimization.
• Collaborate with machine learning engineers to integrate optimized kernels into machine learning frameworks.
• Identify performance bottlenecks, conduct performance profiling, and develop strategies to resolve them.
• Oversee the creation of comprehensive unit tests and ensure the stability of kernel-level code.
• Create and maintain clear documentation for code, APIs, and best practices.
• Stay abreast of the latest developments in kernel development and machine learning to propose innovative solutions.

Requirements:

• Bachelor’s degree in Computer Science, Software Engineering, or a related field.
• Proven experience in kernel development, with a strong 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 leadership skills.
• Self-motivated, detail-oriented, and able to work independently as well as lead a team.
• Experience with GPU programming (CUDA, OpenCL) is a plus.
• Knowledge of operating system internals is a plus.

Tenstorrent offers:

• Highly competitive compensation package and benefits.