Machine Learning Research Engineer

Summary

Join d-Matrix, a leader in generative AI, and become a Machine Learning Research Engineer. Contribute to the R&D of core algorithm-hardware co-design capabilities within our end-to-end solution. Collaborate with product managers and internal software teams to achieve development milestones. Maintain customer-facing software tools and optimize customer workloads for deployment on our AI compute engine. Present progress effectively and work in a fast-paced, collaborative environment. This role offers a hybrid or remote work option in Santa Clara.

Requirements

• Master's degree in Computer Science, Electrical and Computer Engineering, or a related scientific discipline with 1+ years of industry experience
• High proficiency with major deep learning frameworks: PyTorch, TensorFlow
• High proficiency in algorithm analysis, data structure, and Python programming
• Deep, wide and current knowledge in machine learning and modern deep learning applications
• Hands-on experience with CNN, RNN, Transformer neural network architectures in production
• Passionate about AI and thriving in a fast-paced and dynamic startup culture

Responsibilities

• Engage and collaborate with Product Managers to define R&D goals
• Engage and collaborate with internal SW team to meet stack development milestones
• Maintain customer-facing SW tools for customer workload deployment
• Port customer workloads, optimize them for deployment, generate reference implementations and evaluate performance
• Report and present progress timely and effectively

Preferred Qualifications

• Experience with AI HW product development and management
• Experience with specialized HW accelerator systems for deep neural network

Benefits

Hybrid or Remote work option