Senior Neural Network Kernel Software Development Engineer

Targeted Talent; Canada; Remote; CAD 80,000 - 100,000; Full time

Job summary

A technology recruitment agency is seeking a Senior Neural Network Kernel Software Development Engineer. The ideal candidate will have strong C++ skills and experience in neural network optimization, collaborating within agile teams to tackle innovative challenges. This role involves developing efficient implementations tailored to a unique hardware architecture, offering a chance to influence cutting-edge technologies and enjoy substantial benefits.

Benefits

Comprehensive health and extended health benefits

Competitive Salary

Unlimited sick leave

Stock options

Contribution to revolutionizing chip and software technologies

Qualifications

• Profound knowledge of modern C++, focusing on code generation and low-level compute optimizations.
• Familiarity with fundamental Neural Network operator algorithms - Convolutions, Transformers, RNNs.
• Demonstrated capability to navigate challenging problems independently.

Responsibilities

• Write kernels to accelerate Neural Network execution on custom hardware accelerators.
• Design, prototype, and execute low-level C++ programs for neural net operations.
• Define, document, and communicate configuration APIs for kernels to the compiler team.

Skills

C++

Neural Network algorithms

Problem-solving

Python

Education

Degree in Computer Science, Engineering, Math, Physics, or related field (preferably MS or PhD)

Job description

Overview

Senior Neural Network Kernel Software Development Engineer

Our client is making substantial investments in software to enhance the seamless deployment of neural networks on their hardware, streamlining the experience for researchers and developers. The focus involves the optimization of various common neural networks for optimal performance on architectures, facilitated by the software optimization tool flow.

We are seeking software developers who are driven and naturally curious. The chosen candidate will contribute within agile teams, working closely with senior software engineers for mentorship. This role presents an opportunity to tackle novel challenges using cutting-edge technologies, as they build innovative systems from scratch.

As a key team member, you will specialize in constructing efficient implementations of practical neural net kernels tailored to their distinctive hardware architecture. Additionally, you will implement diverse computing algorithms, maximizing computation and communication throughput. This role involves developing a profound understanding of the architecture\'s intricacies, working collaboratively with the architects and compiler engineers.

Responsibilities:

• Experience writing kernels to accelerate Neural Network execution on custom hardware accelerators (not on CPU\'s)
• Design, prototype, and execute low-level, adaptable C++ programs (kernels) for various neural net operations.
• Define, document, and communicate configuration APIs for these kernels to the compiler team.
• Share performance optimization concepts with both compiler engineers and architects working on future product generations.
• Develop comprehensive computation strategies spanning kernels for multichannel and multi-chip neural net implementations.

Requirements:

• Degree in Computer Science, Engineering, Math, Physics, or related field (preferably MS or PhD).
• Profound knowledge of modern C++, with a focus on code generation and low-level compute optimizations.
• Familiarity with fundamental Neural Network operator algorithms - Convolutions, Transformers, RNNs.
• Demonstrated capability to independently navigate challenging, well-defined problems.
• Aptitude and interest in both high-level conceptual understanding and intricate technical details.
• Enthusiasm for problem-solving within highly structured and restricted environments.

Preferred Skills and Experience:

• Proficiency in Python.
• Experience with other AI accelerator programming.
• Strong mathematical aptitude.
• Enjoyment of solving complex problems.

Our client offers:

• Comprehensive health and extended health benefits.
• Competitive Salary
• Unlimited sick leave.
• Stock options.
• Contribution to revolutionizing chip and software technologies with global impact.