Senior Staff ML Researcher LLM Algorithmic Optimization

AI-infrastructure company producing memory-centric hardware, networking, and software for low-latency generative-AI inference in data centers.

Santa Clara, United States
About d-Matrix

d-Matrix develops digital in-memory-compute technology and a full-stack inference platform comprising Corsair accelerators, JetStream networking accelerators, and Aviator software.

View jobs by d-Matrix

Skills

Candidate Availability

Required and preferred rules are kept separate and reflect the wording in the original posting.

About the Role

You will invent, design, and implement efficient algorithms to optimize large language model inference on DNN accelerators. You will apply mathematical, machine learning, algorithmic, and numerical techniques to research involving modern LLM applications.

Requirements

  • MSc or PhD in mathematics, computer science, statistics, physics, or a related STEM field
  • 5+ years of hands-on experience
  • Strong mathematical skills
  • Experience with Python and object-oriented code design

Responsibilities

  • Invent efficient algorithms for large language model inference
  • Design algorithms for DNN accelerators
  • Implement algorithms that optimize large language model inference
  • Apply mathematical and numerical techniques to machine learning research

Benefits

  • Medical insurance
  • Dental insurance
  • Vision insurance
  • 401k
  • Equity
  • Bonus
Senior Staff ML Researcher LLM Algorithmic Optimization at d-Matrix | JobStash