Staff Software Engineer - Kernel

hireVouch
Remote, Canada or US, USA
Remote
Full-time

We are looking for best in class engineers to join our existing top-notch team. When you join us, you will be part of a team that designs, develops and verifies the software that interacts with our chip, collaborating with our hardware engineers and with fellow software engineers in the process.

By creating software that fully realizes the capabilities of the hardware, you will help get AI inference to the general populace.

As part of this exceptional team, you are able to - and get excited by - identifying functional / performance bottlenecks and how to alleviate them in order to achieve scalable and reliable software.

You excel in an environment with complex software and hardware designs.

We are looking for an experienced Senior Kernel Engineer who can help build and optimize our SDK. Our tools and libraries unlock industry-leading performance and power efficiency on our unique at-memory AI inference chips.

We enable customers to compile models directly to run on our architectures, and provide tools to analyze and optimize performance.

The kernel library is at the heart of our SDK, leveraging HW features for fast computations, dividing work flexibly amongst parallel computation elements, as well as providing highly configurable data-flow options for all of our kernels.

The successful candidate will build a deep understanding of the capabilities and limitations of the architecture, and of how features of the kernel library enable performant push-button compilations.

Responsibilities

Efficient and flexible implementation of neural network compute kernels for our chip families

  • Defining / Improving abstractions of our kernel library to accelerate kernel development
  • Analysis and optimization of individual kernel performance and full-network implementation performance
  • Work closely with our compiler and physical allocation teams to enable efficient implementations of networks through our push-button compile tool-flow

Requirements

Computer Science, Engineering, Math, Physics or related degree

  • Experienced in Python, C / C++ and SW design
  • Demonstrated ability to work independently through challenging but tightly constrained problems
  • Demonstrated ability to be a technical leader on projects with teammates or engineers from other teams
  • Interest and ability to work with both high level architectural and very low-level technical details
  • Experience with low-level and / or parallelization optimization, e.g. assembly language development, GPU shaders, SIMD, CUDA, AI inference accelerator kernels

Preferred Skills

Experience with spatial architectures / at-memory compute

  • Knowledge of AI algorithms
  • Strong mathematical skills
  • Enjoy solving very complex problems (like doing IQ tests, solving tricky math problems)
  • 24 days ago
Related jobs
hireVouch
Remote, Canada or US, USA
Remote

When you join us, you will be part of a team that designs, develops and verifies the software that interacts with our chip, collaborating with our hardware engineers and with fellow software engineers in the process. We are looking for an experienced Senior Kernel Engineer who can help build and opt...

StackAdapt
Canada

We're seeking a Staff Software Engineer to help lead our growing backend engineering team. Working with a team of engineers focused on enhancing our Programmatic bidding offering. Promote software development best-practices and conduct rigorous code reviews. At least 5 years experience of software d...

Babylist
Canada

Babylist is currently seeking a dynamic and enthusiastic Staff Software Engineer to join our recently established Health team. In your capacity as a Staff Software Engineer, collaboration with our product and design teams will be integral to shaping new product initiatives and implementing flexible,...

Coinbase
Canada
Remote

As a Staff Software Engineer on our team, you'll play a pivotal role in shaping the future of crypto memberships. You have at least 8 years of experience in software engineering and at least 7 years of experience in developing web apps and shipping user-facing features with JavaScript and modern, co...

hireVouch
Remote, Canada or US, USA
Remote

We operate at the intersection between hardware and software. Our work helps shape both the hardware and software solutions that underlie our AI technology, and we are looking to grow the team through the addition of a scaleout developer. Developing orchestration software solutions to enable such sc...

Coinbase
Canada
Remote

We’re looking for Staff Software Engineers to balance the role of an individual contributor while also anchoring new projects, leading technical direction and mentoring other engineers on the team. Our engineering team leverages a diverse tech stack and our focus is finding smart engineers who’ve wo...

Okta, Inc.
Canada

Existing knowledge of security engineering and application security. ...

Jobber
Canada
Remote

Our Software Engineering team is central to the success of Jobber - focused on creating software that delivers value to tens of thousands of users around the world. As a Staff Software Engineer you will:. A demonstrated level of software engineering experience on a full stack-focused team. You will ...

hireVouch
Remote, Canada and US, USA
Remote

We are investing in software in a big way in order to make it as seamless as possible for researchers and developers to successfully deploy neural networks on our hardware. If you are a self driven generalist capable of working at the intersection of software and hardware then this job might be righ...

StackAdapt
Canada

We're seeking a Staff Engineer to help lead our growing full-stack engineering team. As an Engineer at StackAdapt, you will be directly involved in the development of our advertising platform, producing production level code that will have a direct impact on the success of our product. StackAdapt’s ...