Software Engineer, Senior - AI/ML Workloads

at  dMatrix

Location
Santa Clara, Ca
Type
Full time
Department
R&D - Kernels & Workloads
d-Matrix has fundamentally changed the physics of memory-compute integration with our digital in-memory compute (DIMC) engine. The “holy grail” of AI compute has been to break through the memory wall to minimize data movements. We’ve achieved this with a first-of-its-kind DIMC engine. Having secured over $154M, $110M in our Series B offering, d-Matrix is poised to advance Large Language Models to scale Generative inference acceleration with our chiplets and In-Memory compute approach. We are on track to deliver our first commercial product in 2024. We are poised to meet the energy and performance demands of these Large Language Models. The company has 100+ employees across Silicon Valley, Sydney and Bengaluru.

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 the development and testing infrastructure for next-generation AI hardware. You can build and scale software deliverables in a tight development window. You will work with a team of compiler, ML, and HW architecture experts to build performant ML workloads targeted for d-Matrix’s architecture. You will also research and develop forward looking items that further improve the performance of ML workloads on d-Matrix’s architecture.