AI Software Archtiect Intern at d-Matrix in Santa Clara, CA

Summary: d-Matrix is focused on unleashing the potential of generative AI to power the transformation of technology. They are seeking outstanding computer architects to help accelerate AI application performance at the intersection of both hardware and software. Responsibilities: - As a member of the architecture team, you will contribute to features that power the next generation of inference accelerators in datacenters - This role requires to keep up the latest research in ML architecture and algorithms space, and collaborate with different partner teams including design, verification, and software - Your day-to-day work will include (1) mapping machine learning algorithms into functional, performance specifications of key hardware units (2) proposing new features to enable or accelerate these algorithms, (3) studying the benefits of proposed features with system models (analytical, cycle-level) and generating software references Required Qualifications: - MS with 3+ years of experience or PhD candidates with prior industry internship experience - Solid grasp through academic or industry experience in multiple of the relevant areas – computer architecture, hardware software codesign, performance modeling, ML fundamentals (particularly DNNs) - Self-motivated team player with strong sense of collaboration and initiative Preferred Qualifications: - Hands-on experience with SoC design, bus protocols, memory interfaces, network-on-chip topologies, I/O and multi-device scalable systems - Programming fluency in C/C++ or Python, familiarity with common ML frameworks (e.g. PyTorch) - Research background with publication record in top-tier architecture, or machine learning venues is a huge plus (such as ISCA, MICRO, ASPLOS, HPCA, ICLR, NeurIPS, etc.) Required Skills: Computer architecture, Hardware software co-design, Performance modeling, Machine learning fundamentals Important Skills: C/C++, Python, ML frameworks (e.g. PyTorch), SoC design Nice-to-Have Skills: Team collaboration, Self-motivation