Principal Software Engineer, d-Matrix

AI/ML Kernels

$180-300k

+ Offers Equity + Offers Bonus

Python

Linux

Tensorflow

C++

CUDA

PyTorch

Junior, Mid and Senior 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

21-100 employees

B2BArtificial IntelligenceMachine LearningSaaSAutomation

Open for applications

$180-300k

+ Offers Equity + Offers Bonus

Python

Linux

Tensorflow

C++

CUDA

PyTorch

Junior, Mid and Senior level

San Francisco Bay Area

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

21-100 employees

B2BArtificial IntelligenceMachine LearningSaaSAutomation

Company mission

To deliver best-in-class economics for AI inference acceleration in the datacenter.

Job

Company

Role

Who you are

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
MS in Computer Engineering, Math, Physics or related degree with 3+ years of industry experience or PhD in Computer Engineering, Math, Physics or 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 environment 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 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
Work experience at a cloud provider or AI compute / sub-system 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 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

The explosion in the field of AI puts significant strain on traditional computers, and the silicon architectures currently in use are a bottleneck on the speed and effectiveness of emerging AI models. d-Matrix aims to tackle this challenge by with its digital in-memory computing accelerator for AI. Using a combination of machine learning software tools and modular chiplets it delivers a huge increase in AI computing efficiency without a corresponding increase in energy use.

D-Matrix has major competitors in the race to deliver the next generation of scalable AI solutions. These include the likes of Nvidea which is developing GPU architecture that runs at ten times the efficiency of existing models. However the flexibility and easy scalability of d-Matrix's solution helps it stand out, and its novel approach has gained a high degree of support from investors. The company is currently expanding its operations in the US, India and Australia, and is well poised for success as demand for AI compute solutions continues to grow exponentially.

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 HQ

Santa Clara, CA

Leadership

Sid Sheth

(CEO)

Previously Director at NetLogic Microsystems, Director of Marketing at Broadcom Inc and SVP/GM atInphi Corporation.

Sudeep Bhoja

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