GrepJob
d-Matrix

Principal Software Engineer - Kernels

d-Matrix
Apply
about 3 hours ago
Santa Clara, CA, USAStaff+
H1B Sponsor

Base Salary

$200k - $300k/yr

Responsibilities

  • Develop and maintain software kernels for AI hardware.
  • Enhance the software stack for the AI compute engine.
  • Collaborate with compiler experts and other software and hardware teams.
  • Optimize hardware-software co-design for performance and efficiency.
  • Map computational graphs from AI frameworks to hardware architectures.

Requirements

  • MS in computer engineering, math, physics, or related field with 12+ years of experience, or PhD with 7+ years.
  • Strong understanding of computer architecture and machine learning fundamentals.
  • Proficient in C/C++ and Python in Linux environments.
  • Experience with specialized hardware like FPGAs, DSPs, and GPUs.
  • Familiarity with ML frameworks such as TensorFlow or PyTorch is preferred.