Staff Machine Learning Research Scientist

D-matrix Corporation - San Diego, CA

Full-time - Mid Level

San Diego, CA

51-100 employees

Professional, Scientific, and Technical Services

About the position

The Machine Learning Team at d-Matrix is focused on the research and development of core algorithm-hardware co-design capabilities for the company's innovative AI compute engine. This role involves designing and implementing efficient deep learning techniques tailored for d-Matrix's technology, collaborating with both internal teams and external researchers, and optimizing workloads for real-world AI applications. The position offers the opportunity to contribute to cutting-edge advancements in AI and engage with top academic labs.

Responsibilities

• Design, implement and evaluate efficient deep neural network architectures and algorithms for d-Matrix's AI compute engine.
• Engage and collaborate with internal and external ML researchers to meet R&D goals.
• Collaborate with the software team to meet stack development milestones.
• Conduct research to guide hardware design.
• Develop and maintain tools for high-level simulation and research.
• Port customer workloads, optimize them for deployment, generate reference implementations and evaluate performance.
• Report and present progress timely and effectively.
• Contribute to publications of papers and intellectual properties.

Requirements

• Master's degree in Computer Science, Electrical and Computer Engineering, or a related technical discipline with 3+ years of industry experience, PhD preferred with 1+ year of industry experience.
• High proficiency with major deep learning frameworks: PyTorch is a must.
• High proficiency in algorithm analysis, data structure, and Python programming is a must.

Nice-to-haves

• Proficiency with C/C++ programming is preferred.
• Proficiency with GPU CUDA programming is preferred.
• Deep, wide and current knowledge in machine learning and modern deep learning is preferred.
• Experience in real-world data science projects in an industry setting is preferred.
• Experience with efficient deep learning techniques such as quantization, sparsity, and distillation is preferred.
• Experience with specialized hardware accelerator systems for deep neural networks is preferred.

Benefits

• Hybrid work environment with onsite work at Santa Clara, CA headquarters or San Diego, CA 3 days per week.