Informacje o stanowisku

Join the exciting world of Artificial Intelligence as an Acceleration Kernel Developer at Tenstorrent. You will play a crucial role in Tenstorrent as a kernel developer by optimizing low-level workloads, kernel development, and enhancing our softwares 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 Warsaw, Poland.

Responsibilities

• Kernel Development: Take initiative and 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.
• 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.
• Teamwork: Collaborate with a team of engineers and software developers on projects to achieve shared goals.

Experience & Qualifications

• Bachelor’s degree in Computer Science, Computer 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 teamwork skills.
• Self-motivated, detail-oriented, and able to work independently as well as in a team.
• Comfortable working in a Linux environment.
• Experience with GPU programming (CUDA, OpenCL) is a plus.
• Experience with VMs, Docker, etc.
• Familiarity with Atlassian, Git, VS is a plus.
• Knowledge of operating system internals is a plus.

Tenstorrent offers a highly competitive compensation package and benefits, and we are an equal opportunity employer.

Warszawa - Oferty pracy w okolicznych lokalizacjach