Staff Software Engineer, AI Kernels & Compiler Tech

d-Matrix; Santa Clara (CA); Hybrid; USD 120,000 - 180,000; Full time

Job summary

d-Matrix is seeking a Staff Software Engineer - Kernels to join our team in Santa Clara, California. In this hybrid role, you'll focus on productizing the software stack for our AI compute engine. Candidates should have at least 5 years of experience (1 year for PhD holders) in software development, particularly with software kernels for hardware architectures. Proficiency in C/C++ and Python, along with a solid understanding of machine learning fundamentals, is essential. We value self-motivated team players with a strong sense of ownership and leadership.

Qualifications

• 5+ years of industry experience or 1+ years with a PhD.
• Strong grasp of data structures and system software.
• Experience with development for embedded SIMD vector processors.
• Experience in implementing ML workloads, GEMMs, and Convolutions.

Responsibilities

• Develop and maintain software kernels for AI hardware.
• Work with compiler experts to build compiler infrastructure.
• Map computational graphs to architecture.
• Optimize hardware-software co-design aspects.

Skills

C/C++ development

Python development

Computer architecture

Machine learning fundamentals

Linux environments

Education

MS in computer engineering, math, physics, or related

PhD in computer engineering, math, physics, or related

Tools

CUDA

ML frameworks

TensorFlow

PyTorch

Job description

d-Matrix is seeking a Staff Software Engineer - Kernels to join our team in Santa Clara, California. In this hybrid role, you'll focus on productizing the software stack for our AI compute engine. Candidates should have at least 5 years of experience (1 year for PhD holders) in software development, particularly with software kernels for hardware architectures. Proficiency in C/C++ and Python, along with a solid understanding of machine learning fundamentals, is essential. We value self-motivated team players with a strong sense of ownership and leadership.