Senior Neural Network Kernel Software Development Engineer

Job Description

Job Description

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 :

Preferred Skills and Experience :

Our client offers :