Senior ML Researcher: LLM Optimization (Hybrid)

d-Matrix; California (MO); Hybrid; USD 180,000 - 240,000; Full time

Job summary

d-Matrix is seeking a Sr. Staff ML Researcher to develop and optimize algorithms for LLM inference on our DNN accelerators. You will join a team of mathematicians, ML researchers, and engineers to push the boundaries of ML and AI applications.

Based in Santa Clara, CA, this hybrid role combines deep theoretical work with practical implementation, requiring a strong mathematical background and hands-on Python/OOP experience. Transformer knowledge is a plus.

Qualifications

• MSc or PhD in math, CS, statistics, physics, or related STEM with 5+ years hands-on experience.
• Experience in Python and OOP code design.
• Experience with transformer architecture is advantageous but not mandatory.

Responsibilities

• Invent, design, and implement efficient algorithms for optimizing LLM inference on DNN accelerators.
• Work with a close-knit team of mathematicians, ML researchers, and ML engineers on advanced algorithms and numerical techniques.

Skills

Strong math skills

Python

OOP design

Transformer architectures

Education

MSc/PhD in math, CS, stats, physics or related STEM

Job description

d-Matrix is seeking a Sr. Staff ML Researcher to develop and optimize algorithms for LLM inference on our DNN accelerators. You will join a team of mathematicians, ML researchers, and engineers to push the boundaries of ML and AI applications.

Based in Santa Clara, CA, this hybrid role combines deep theoretical work with practical implementation, requiring a strong mathematical background and hands-on Python/OOP experience. Transformer knowledge is a plus.