4 months ago
Base Salary
$195k - $285k/yr
Responsibilities
- Develop, enhance, and maintain software kernels for next-generation AI hardware.
- Map algorithms and AI-framework computational graphs to specialized hardware architectures.
- Optimize hardware-software co-design tradeoffs across the full-stack toolchain.
- Implement ML operators including GEMMs, convolutions, BLAS, SIMD operations, softmax, layer normalization, and pooling.
- Collaborate with compiler experts to build compiler infrastructure and work with ML, systems, DSP, CPU, and hardware teams.
- Build and scale software deliverables within tight development windows.
Requirements
- Master’s degree in computer engineering, mathematics, physics, or a related field plus 10+ years of industry experience, or a PhD in one of these fields plus 5+ 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 for specialized hardware including FPGAs, DSPs, GPUs, and AI accelerators.
- Experience developing for embedded SIMD vector processors such as Tensilica.
- Strong ownership, leadership, self-motivation, and teamwork.
- Preferred experience with startups, small teams, or incubation environments.
- Preferred experience with TensorFlow, PyTorch, ML compilers, MLIR, LLVM, TVM, Glow, and ML models for computer vision, NLP, or recommendation.
- Preferred experience at a cloud provider or AI compute/subsystem company.
Benefits
- Hybrid work arrangement requiring onsite work at the Santa Clara, California headquarters at least 3 days per week.
- Equal opportunity and affirmative action workplace.
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.