About the position
At d-Matrix, we are focused on unleashing the potential of generative AI to power the transformation of technology. We are at the forefront of software and hardware innovation, pushing the boundaries of what is possible. Our culture is one of respect and collaboration. We value humility and believe in direct communication. Our team is inclusive, and our differing perspectives allow for better solutions. We are seeking individuals passionate about tackling challenges and are driven by execution. Ready to come find your playground? Together, we can help shape the endless possibilities of AI. d-Matrix is seeking a Machine Learning Engineer to join our Algorithm Team. We're looking for someone to invent, design, and implement efficient algorithms that will be used to optimize Large Language Model inference on DNN Accelerators we develop. You would be part of a close-knit team of mathematicians, ML researchers, and ML engineers who create and apply advanced algorithmic and numerical techniques to the most cutting-edge and high-impact research in the overlap of mathematics, ML, and modern LLM applications.
Responsibilities
- Implement advanced quantization algorithms for modern LLMs
- Support and develop numerical libraries
- Participate in design of the next generation chips for multi-model training and inference.
Requirements
- MSc or PhD in CS, statistics, physics, or a related STEM field
- 4+ years of experience in industrial coding and OOP design
- High motivation to innovate and collaborate with experts from various fields
- Experience in Python
Nice-to-haves
- Experience with transformer architecture is advantageous but not mandatory.
