Senior Staff ML Researcher (LLM Algorithmic Optimization)

Join D-Matrix as a Senior Staff ML Researcher, where you will invent, design, and implement efficient algorithms to optimize large language model inference on our DNN accelerators. You will be part of a close-knit team of mathematicians, ML researchers, and ML engineers, working on cutting-edge research in the overlap of mathematics, ML, and modern LLM applications. Enjoy benefits such as equity, healthcare, flexible time-off, paid paternity leave, a 401k retirement plan, and a remote & hybrid working model.

• Invent, design, and implement efficient algorithms for optimizing large language model inference on DNN accelerators.
• Collaborate with a team of mathematicians, ML researchers, and ML engineers to apply advanced algorithmic and numerical techniques.
• Contribute to cutting-edge research in the overlap of mathematics, machine learning, and modern LLM applications.
• Experience in Python and OOP code design
• Strong mathematical skills—MSc or PhD in math, CS, statistics, physics, or a related STEM field and 5+ years of hands-on experience
• Experience with transformer architecture is advantageous but not mandatory