d-Matrix

Senior Staff Software Engineer – SIMD Kernels

d-Matrix

full-time

Posted on:

Location Type: Hybrid

Location: Santa Clara • California • 🇺🇸 United States

Visit company website
✨ AI Apply
Apply

Salary

💰 $190,000 - $300,000 per year

Job Level

Senior

Tech Stack

LinuxPython

About the role

  • Help productize the SW stack for AI compute engine.
  • Responsible for the development, enhancement, and maintenance of software kernels for next-generation AI hardware.
  • Map algorithms to various hardware architectures.
  • Collaborate with a team of compiler experts and other software (ML, Systems) and hardware (mixed signal, DSP, CPU) experts to build out compiler infrastructure.
  • Optimize and trade-off various aspects of hardware-software co-design.
  • Build and scale software deliverables in a tight development window.

Requirements

  • MS or PhD in computer engineering, math, physics, or a related degree with 10+ years of industry experience.
  • Strong grasp of computer architecture, data structures, system software, and machine learning fundamentals.
  • Proficient in C/C++ and Python development in Linux environment and using standard development tools.
  • Experience implementing algorithms in high-level languages such as C/C++ and Python.
  • Experience implementing algorithms for specialized hardware such as FPGAs, DSPs, GPUs, AI accelerators using libraries such as CUDA, etc.
  • Experience in implementing operators commonly used in ML workloads—GEMMs, Convolutions, BLAS, SIMD operators for operations like softmax, layer normalization, pooling, etc.
  • Experience with development for embedded SIMD vector processors such as Tensilica.
  • Self-motivated team player with a strong sense of ownership and leadership.
Benefits
  • Offers Equity
  • Offers Bonus
  • Medical/Dental/Vision/401k
  • Diverse package of benefits centered around the well-being of our employees and their loved ones.

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard skills
CC++PythonLinuxalgorithmsmachine learningcomputer architecturedata structuresFPGAsDSPs
Soft skills
self-motivatedteam playerownershipleadership