Software Engineer, Senior Staff - SIMD Kernels (25-357) jobs in United States
cer-icon
Apply on Employer Site
company-logo

d-Matrix · 2 weeks ago

Software Engineer, Senior Staff - SIMD Kernels (25-357)

positionSanta Clara, CA
timeFull-time
remoteHybrid
senioritySenior Level, Lead/Staff
date10+ years exp

d-Matrix is focused on unleashing the potential of generative AI to power the transformation of technology. The Senior Staff Software Engineer will help productize the software stack for their AI compute engine by developing and maintaining software kernels for machine learning operators.

AI InfrastructureArtificial Intelligence (AI)Cloud InfrastructureData CenterSemiconductor
check
H1B Sponsor Likelynote

Responsibilities

Develop, enhance, and maintain software kernels for ML operators—such as softmax, layer norm, and activation functions—for next‑generation AI hardware
Build solutions that make our SDK intuitive for developers to use and analyze performance
Deliver and scale high‑quality software quickly in a fast‑moving development environment

Qualification

C/C++ developmentPython developmentComputer architectureMachine learning fundamentalsML frameworksAlgorithms for specialized hardwareSoftware kernelsTeam playerLeadershipSelf-motivated

Required

MS or PhD in computer engineering, math, physics, or a related degree with 10+ 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 environment and using standard development tools
Experience in implementing algorithms in C/C++ and Python
Experience in implementing algorithms for specialized hardware such as FPGAs, DSPs, GPUs, AI accelerators using libraries such as CUDA, etc
Experience in implementing operators commonly used in ML workloads—GEMMs, Convolutions, softmax, layer normalization, pooling, etc
Self-motivated team player with a strong sense of ownership and leadership

Preferred

Prior startup, small team, or incubation experience
Experience with ML frameworks such as TensorFlow and/or PyTorch
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
Experiences in development for embedded SIMD vector processors such as Tensilica
Work experience at a cloud provider or AI compute/subsystem company

Company

d-Matrix

twittertwittertwitter
company-logo
D-Matrix is a platform that enables data centers to handle large-scale generative AI inference with high throughput and low latency.
dateFounded in 2019
location
Santa Clara, California, USA
people-size51-200 employees
web-link
https://www.d-matrix.ai

H1B Sponsorship

d-Matrix has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (20)
2024 (15)
2023 (8)
2022 (7)

Funding

Current Stage
Growth Stage
Total Funding
$429M
Key Investors
Temasek HoldingsTSVC
2025-11-12Series C· $275M
2023-09-06Series B· $110M
2022-04-20Series A· $44M

Leadership Team

leader-logo
Peter Buckingham
Senior Vice President, Software Engineering
linkedin

Recent News

Crowdfund Insider
AI Adoption Trends : 2025 Saw Emergence of 1000+ Agentic AI Offerings
2025-12-22
CB Insights
AI meets the physical economy: November mega-rounds pivot to industrial AI
2025-12-17
Crunchbase News
Jeff Bezos’s Project Prometheus Joins The Unicorn Board Alongside 18 Other Startups In November
2025-12-10
Company data provided by crunchbase