Senior Staff Software Engineer, d-Matrix
AI/ML Kernels
$180-300k
+ Equity + Bonus
AI computing and efficiency software
Job no longer available
AI computing and efficiency software
201-500 employees
Job no longer available
$180-300k
+ Equity + Bonus
201-500 employees
Company mission
To unleash the potential of generative AI to power the transformation of technology.
Role
Who you are
- MS in computer engineering, math, physics, or a related degree with 7+ years of industry experience or PhD in computer engineering, math, physics, or a related degree with 1+ 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 environments 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, and 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
Desirable
- Prior startup, small team, or incubation experience
- Experience with ML frameworks such as TensorFlow and/orPyTorch
- Experience working with ML compilers and algorithms, such as MLIR, LLVM, TVM, Glow, etc
- Experience with a deep learning framework (such as PyTorch or TensorFlow) and ML models for CV, NLP, or recommendation
- Work experience at a cloud provider or AI compute/subsystem company
What the job involves
- The role requires you to be part of the team that helps productize the SW stack for our AI compute engine
- As part of the software team, you will be responsible for the development, enhancement, and maintenance of software kernels for next-generation AI hardware
- You possess experience building software kernels for HW architectures
- You possess a very strong understanding of various hardware architectures and how to map algorithms to the architecture
- You understand how to map computational graphs generated by AI frameworks to the underlying architecture
- You have had past experience working across all aspects of the full-stack toolchain and understand the nuances of what it takes to optimize and trade off various aspects of hardware-software co-design
- You can build and scale software deliverables in a tight development window
- You will work with a team of compiler experts to build out the compiler infrastructure, working closely with other software (ML, systems) and hardware (mixed signal, DSP, CPU) experts in the company
Our take
d-Matrix is redesigning how AI gets done, tackling the limit f traditional chips with its Corsair platform. Using digital in-memory computing and a chiplet-based design, it delivers fast, power-efficient AI interface that's built for today's data-heavy models, without the energy overload.
Recently, the company has been making some serious moves. Its secured substantial funding, with backing from Microsoft, and partnering with HTEC to boost its software capabilities. With Corsair already showing serious speed gains over GPUs like Nvidia's H100, d-Matrix is proving it can deliver real-world performance at scale.
But instead of competing head-on with GPU giants, d-Matrix is focusing on where AI meets people. And with AI growing fast, especially for interactive and generative applications, d-Matrix is building the hardware that helps AI keep up with this demand.
Freddie
Company Specialist at Welcome to the Jungle
Company
Funding (2 rounds)
Sep 2023
$110m
SERIES B
Apr 2022
$44m
SERIES A
Total funding: $154m
Company benefits
- Equity
- Health care
- Flexible time-off
- Paid paternity leave
- 401k retirement plan
- Remote & hybrid working model
- Work with world-leading engineers and researchers
- Peer coaching & development
- Team-building activities
- Happy hours
- Free food and snacks
- On-site fitness center
Company HQ
Santa Clara, CA
Leadership
Sid Sheth
(Co-Founder, CEO & President)Previously Director at NetLogic Microsystems, Director of Marketing at Broadcom Inc and SVP/GM at Inphi Corporation.
Sudeep Bhoja
(Co-Founder & CTO)Previously CTO at Inphi Corporation, Technical Director at Broadcom and Chief Architect at Big Bear Networks.
Salary benchmarks
We don't have enough data yet to provide salary benchmarks for this role.
Submit your salary to help other candidates with crowdsourced salary estimates.
Share this job
View 23 more jobs at d-Matrix