NN Kernel Engineer

NN Kernel Engineer

Untether AI

Toronto
Permanent
Competitive
Share

We are looking for an experienced Kernel Engineer. The objective of the role is to build efficient implementations of real-world neural nets kernels specialized for our unique architecture, including implementation of other algorithms like beam search, and designing layers to maximize layer-to-layer and IO throughput. The successful candidate will build a deep understanding of the capabilities, limitations and details of our architecture and work closely with our architects and compiler engineers.

developer