5 months ago
Base Salary
$200k - $300k/yr
Responsibilities
- Develop, enhance, and maintain software kernels for next-generation AI hardware.
- Help productize the software stack for the company’s AI compute engine.
- Map algorithms and computational graphs generated by AI frameworks to underlying hardware architectures.
- Optimize and make tradeoffs across the hardware-software co-design and full-stack toolchain.
- Build and scale software deliverables within tight development windows.
- Collaborate with compiler experts on compiler infrastructure and with machine learning, systems, mixed-signal, DSP, CPU, and hardware teams.
Requirements
- MS in computer engineering, mathematics, physics, or a related field with 12+ years of industry experience, or a PhD in one of these fields with 7+ years of industry experience.
- Strong understanding of computer architecture, data structures, system software, and machine learning fundamentals.
- Proficiency in C/C++ and Python development in Linux environments using standard development tools.
- Experience implementing algorithms in C/C++ and Python and for specialized hardware such as FPGAs, DSPs, GPUs, and AI accelerators.
- Experience implementing ML operators including GEMMs, convolutions, BLAS, SIMD operations, softmax, layer normalization, and pooling.
- Experience developing for embedded SIMD vector processors such as Tensilica.
- Experience with ML frameworks such as TensorFlow or PyTorch is preferred.
- Experience with ML compilers and algorithms such as MLIR, LLVM, TVM, or Glow is preferred.
- Experience with deep learning frameworks and ML models for computer vision, natural language processing, or recommendation is preferred.
- Startup, small-team, incubation, cloud-provider, or AI-compute/subsystem-company experience is preferred.
- Strong ownership, leadership, collaboration, and self-motivation.
Benefits
- Hybrid work arrangement with 3–5 days per week onsite at the Santa Clara, California headquarters.
About d-Matrix
d-Matrix builds AI inference computing platforms for data centers, combining custom silicon with systems, networking, and software. Its flagship Corsair platform and JetStream fabric focus on low-latency, energy-efficient generative AI inference at scale. Founded in 2019 and headquartered in Santa Clara, California, the privately held company sells hardware with accompanying software to cloud providers and enterprises deploying large AI models.