d-Matrix is focused on unleashing the potential of generative AI to transform technology, fostering a culture of respect and collaboration. They are seeking a Principal Software Engineer - Kernels to develop, enhance, and maintain software kernels for next-generation AI hardware, while collaborating closely with experts across software and hardware domains.
Responsibilities:
- The role requires you to be part of the team that helps productize the SW stack for our AI compute engine
- As part of the software team, you will be responsible for the development, enhancement, and maintenance of software kernels for next-generation AI hardware
- You possess experience building software kernels for HW architectures
- You possess a very strong understanding of various hardware architectures and how to map algorithms to the architecture
- You understand how to map computational graphs generated by AI frameworks to the underlying architecture
- You have had past experience working across all aspects of the full-stack toolchain and understand the nuances of what it takes to optimize and trade off various aspects of hardware-software co-design
- You can build and scale software deliverables in a tight development window
- You will work with a team of compiler experts to build out the compiler infrastructure, working closely with other software (ML, systems) and hardware (mixed signal, DSP, CPU) experts in the company