Senior Staff of AI / Machine Learning System Software Engineer, d-Matrix

$180-280k

+ Offers Equity + Offers Bonus; We also offer incentive opportunities that reward employees based on individual and company performance

Python

Linux

Tensorflow

C++

Kubeflow

PyTorch

Senior and 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-280k

+ Offers Equity + Offers Bonus; We also offer incentive opportunities that reward employees based on individual and company performance

Python

Linux

Tensorflow

C++

Kubeflow

PyTorch

Senior and 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

BS in Computer Science, Engineering, Math, Physics or related degree with 8+ years of industry software development experience and MS in Computer Science, Engineering, Math, Physics or related degree preferred with 5 + years
Strong grasp of computer architecture, data structures, system software, and machine learning fundamentals
Proficient in C/C++/Python development in Linux environment and using standard development tools
Experience with distributed, high performance software design and implementation
Self-motivated team player with a strong sense of ownership and leadership

Desirable

MS or PhD in Computer Science, Electrical Engineering, or related fields
Prior startup, small team or incubation experience
Work experience at a cloud provider or AI compute / sub-system company
Experience implementing SIMD algorithms on vector processors
Experience with open source ML compiler frameworks such as MLIR
Experience with deep learning frameworks (such as PyTorch, Tensorflow)
Experience with deep learning runtimes (such as ONNX Runtime, TensorRT,…)
Experience with inference servers/model serving frameworks (such as Triton, TFServ, KubeFlow,…)
Experience with distributed systems collectives such as NCCL, OpenMPI
Experience deploying ML workloads on distributed systems, in a multitenancy environment
Experience with MLOps from definition to deployment including training, quantization, sparsity, model preprocessing, and deployment
Experience training, tuning and deploying ML models for CV (ResNet,..), NLP (BERT, GPT), and/or Recommendation Systems (DLRM)

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 the next-generation AI Deployment software
You have had past experience working across all aspects of the full stack tool chain and understand the nuances of what it takes to optimize and trade-off various aspects of hardware-software co-design
You are able to 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 14 more jobs at d-Matrix