D Matrix

Principal Software Engineer, SDK & Lowering Stack

Santa Clara, Remote. Posted 3 months ago

The role

You will architect and develop a kernel authoring SDK targeting the Corsair dataflow architecture. You will also lead junior engineers and build software deliverables for the software stack.

What you will do

  • Architect and develop the kernel authoring SDK targeting the Corsair dataflow architecture
  • Champion software engineering best practices including CI/CD pipelines, automated testing, and code review processes
  • Build intuitive, domain-specific APIs
  • Create development tools for kernel authoring and modeling lowering flows
  • Lead and mentor junior engineers
  • Leverage systems engineering to understand hardware and software interactions
  • Take ownership of delivering solutions by interacting with stakeholders, generating requirements, documenting designs, implementing, debugging, and deploying
  • Build and scale software deliverables in a tight development window

Required qualifications

  • MS in Computer Engineering, Math, Physics, or a related degree with 8 years of industry experience or a PhD in Computer Engineering, Math, Physics, or a related degree with 4 years of industry experie
  • Proficient in C/C++ and Python development in a Linux environment and using standard development tools
  • Proven track record of driving software engineering best practices, including CI/CD integration, continuous testing, and leading comprehensive code reviews
  • Deep grasp of computer architecture, data structures, system software, and machine learning fundamentals
  • Demonstrated experience designing high-quality, scalable APIs and SDKs, with a comprehensive systems engineering mindset to solve end-to-end problems
  • Self-motivated team player with a profound sense of ownership and leadership

Nice to have

  • Proven experience writing and optimizing high-performance kernels targeting GPUs, TPUs, or custom ML accelerators
  • Background working in High-Performance Computing (HPC) environments, managing large-scale distributed workloads and resource optimization
  • Experience developing ML compilers and working with intermediate representations or algorithms (e.g., MLIR, LLVM, TVM, Glow, etc.)
  • Familiarity with leveraging AI-assisted coding tools to accelerate software development and elevate code quality
  • Prior startup, small team, or incubation experience
  • Work experience at a cloud provider or AI compute/subsystem company

About the company

d-Matrix has pioneered a memory-centric architecture that brings compute directly into memory. The team works on chips, software, and rack-scale systems to tackle AI infrastructure challenges.

Principal Software Engineer, SDK & Lowering Stack at D Matrix | JobZoey