Ai Software Architect, Principal - Santa Clara, United States - d-Matrix

d-Matrix

Verified Company

Santa Clara, United States

3 weeks ago

Posted by:

Mark Lane

beBee recruiter

Description

Location

Santa Clara, Ca

Type

Full time

Department

R&D
CTO & Architecture

Compensation

IC6$180K $300K
Offers Equity
Offers Bonus
The pay range below is for all roles at this level across all US locations and functions. Individual pay rates depend on a number of factors—including the role's function and location, as well as the individual's knowledge, skills, experience, education, and training. We also offer incentive opportunities that reward employees based on individual and company performance. This is in addition to our diverse package of benefits centered around the wellbeing of our employees and their loved ones. In addition to the usual Medical/Dental/Vision/401k, our inclusive rewards plan empowers our people to care for their whole selves. An investment in your future is an investment in ours.

_d-Matrix has fundamentally changed the physics of memory-compute integration with our digital in-memory compute (DIMC) engine. The "holy grail" of AI compute has been to break through the memory wall to minimize data movements. We've achieved this with a first-of-its-kind DIMC engine. Having secured over $154M, $110M in our Series B offering, d-Matrix is poised to advance Large Language Models to scale Generative inference acceleration with our chiplets and In-Memory compute approach. We are on track to deliver our first commercial product in 2024. We are poised to meet the energy and performance demands of these Large Language Models._
_The company has 100+ employees across Silicon Valley, Sydney and Bengaluru. _

Location:

Hybrid, working onsite at our Santa Clara, CA headquarters 3 days per week.

The role:
Principal AI SW Architect

What you will do:

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 daytoday work will include (1) mapping machine learning workloads into functional, performant algorithms using dMatrix 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

What you will bring:

Minimum:

MS, PHD, MSEE with 8+ years of experience or PhD with 5+ 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 toptier architecture, or machine learning venues is a huge plus (such as ISCA, MICRO, ASPLOS, HPCA, ICLR, NeurIPS, etc.)
Selfmotivated team player with strong sense of collaboration and initiative

Equal Opportunity Employment Policy

d-Matrix is proud to be an equal opportunity workplace and affirmative action employer. We're committed to fostering an inclusive environment where everyone feels welcomed and empowered to do their best work. We hire the best talent for our teams, regardless of race, religion, color, age, disability, sex, gender identity, sexual orientation, ancestry, genetic information, marital status, national origin, political affiliation, or veteran status. Our focus is on hiring teammates with humble expertise, kindness, dedication and a willingness to embrace challenges and learn together every day.