Sr. Staff ML Researcher, LLM Algorithmic Optimization
d-Matrix, Santa Clara
About the job
Join d-Matrix and help build the future of AI inference. We’ve pioneered a memory-centric architecture that brings compute directly into memory, making AI faster, more energy efficient, and accessible on a global scale. From chips and software to rack-scale systems, our team is tackling some of AI infrastructure’s hardest challenges — and shipping breakthrough technology today. If you want to create what comes next and help make infinite inference possible, we’d love to hear from you.
Our culture is built on respect, collaboration, and direct communication. We value humility, we believe our different perspectives lead to better solutions, and we’re looking for people who are driven by execution and energized by hard problems.
Role Overview
d-Matrix is seeking a Senior Staff ML Researcher to join our Algorithms team and develop new methods for running large language models efficiently on our DNN accelerators. This is a hands-on research role at the intersection of machine learning, applied mathematics, numerical computing, and AI hardware. You will invent, evaluate, and implement algorithmic techniques that improve LLM inference performance while preserving model quality, with a focus on challenges such as numerical precision, model compression, memory efficiency, and efficient execution on accelerator hardware. You will work closely with mathematicians, ML researchers, ML engineers, compiler engineers, and hardware architects to translate research ideas into practical solutions for cutting-edge generative AI workloads.