Staff Software Engineer SIMD Kernels
AI-infrastructure company producing memory-centric hardware, networking, and software for low-latency generative-AI inference in data centers.
Funding history
About d-Matrix
d-Matrix develops digital in-memory-compute technology and a full-stack inference platform comprising Corsair accelerators, JetStream networking accelerators, and Aviator software.
Skills
Candidate Availability
Required and preferred rules are kept separate and reflect the wording in the original posting.
About the Role
You will productize the AI compute software stack by developing, enhancing, and maintaining kernels for ML operators. You will map algorithms and framework computational graphs to modern hardware architectures, improve SDK usability and performance analysis, and navigate hardware-software co-design trade-offs across the toolchain.
Requirements
- MS or PhD in computer engineering, math, physics, or a related field
- 5+ years of industry experience
- Computer architecture knowledge
- Data structures knowledge
- System software knowledge
- Machine learning fundamentals
- C/C++ development
- Python development
- Linux development
- Algorithm implementation
- Specialized hardware development
- Machine learning operator implementation
Responsibilities
- Develop, enhance, and maintain software kernels for machine learning operators
- Build SDK solutions for developer usability and performance analysis
- Map algorithms and AI framework computational graphs onto hardware architectures
- Work across the toolchain to resolve hardware-software co-design trade-offs
- Deliver scalable, high-quality software
Benefits
- Equity
- Bonus
- Medical insurance
- Dental insurance
- Vision insurance
- 401(k)
- Incentive opportunities