Remote Staff Software Engineer - Kernel

Summary

Join Untether AI as a Senior Kernel Engineer to help build and optimize our SDK, leveraging your expertise in neural network compute kernels, kernel library abstractions, and performance optimization.

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

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

Preferred Qualifications

• Experience with spatial architectures / at-memory compute
• Knowledge of AI algorithms
• Strong mathematical skills

Benefits

• 20 vacation days
• Strong health and extended health benefits
• Unlimited sick days
• Stock options