Principal AI System Architect, d-Matrix

$180-300k

Plus equity and bonus

Python

C++

PyTorch

Expert level

San Francisco Bay Area

3 days a week in office (Santa Clara, CA)

AI computing and efficiency software

Open for applications

AI computing and efficiency software

201-500 employees

B2BArtificial IntelligenceMachine LearningSaaSAutomation

Open for applications

$180-300k

Plus equity and bonus

Python

C++

PyTorch

Expert level

San Francisco Bay Area

3 days a week in office (Santa Clara, CA)

201-500 employees

B2BArtificial IntelligenceMachine LearningSaaSAutomation

Company mission

To unleash the potential of generative AI to power the transformation of technology.

Job

Company

Role

Who you are

We are seeking individuals passionate about tackling challenges and are driven by execution
MS, PHD, MSEE with 12+ years of experience or PhD with 8+ years of applicable experience
Solid grasp through academic or industry experience in multiple of the relevant areas – computer architecture, hardware software codesign, CPU/GPU kernel development, ML fundamentals (particularly DNNs)
Programming fluency in C/C++, Python and familiarity with common ML frameworks (e.g. PyTorch) is preferred. Prior experience developing algorithms for GPUs or DSPs is desired
Research background with publication record in top-tier architecture, or machine learning venues is a huge plus (such as ISCA, MICRO, ASPLOS, HPCA, ICLR, NeurIPS, etc.)
Self-motivated team player with strong sense of collaboration and initiative

What the job involves

D-Matrix is seeking outstanding software architects to help accelerate AI application performance at the intersection of both hardware and software, with particular focus on hardware-aware algorithms, deep optimizations, bottleneck analysis for emerging workloads (such as generative inference etc.)
Our acceleration philosophy cuts through the system ranging from efficient computing and data movements along with co-design of dataflow and distributed systems manifested in scalable, high performance algorithms
As a member of the architecture team, you will contribute to the definition of key software stack components while also feeding forward into the next hardware generation
This role requires to keep up the latest research in ML architecture and algorithms space, and collaborate with different partner teams including software and hardware engineering
Your day-to-day work will include (1) mapping machine learning workloads into functional, performant algorithms using d-Matrix architecture (2) proposing new features to enable or accelerate these algorithms, (3) studying the benefits of proposed features with performance models and generating software references

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 14 more jobs at d-Matrix