AI/ML Hardware Architect Intern - Remote

Overview

The AGI (Artificial General Intelligence) Computing Lab is dedicated to solving the complex system-level challenges posed by the growing demands of future AI/ML workloads. Our team is committed to designing and developing scalable platforms that can effectively handle the computational and memory requirements of these workloads while minimizing energy consumption and maximizing performance.

In Short

• Research novel architectures for high-performance deep learning applications.
• Research design space exploration methodology and pruning for efficient architecture exploration.
• Design algorithms and microarchitectural features to optimize data locality to minimize energy consumption.
• Work closely with the compiler team to integrate new ML techniques and algorithms into the compiler.
• Collaborate with cross-functional teams to define and deliver discover and implement microarchitecture features and improvements.
• Contribute to ML architecture research and write research papers.
• Stay up-to-date with the latest trends and advancements in the field of ML architectures.
• Complete other responsibilities as assigned.

Requirements

• Pursuing Masters, or PhD in Computer Science or Electrical Engineering preferred.
• Research experience with hardware architectures such as CPUs, GPUs, TPUs, and NPUs.
• Published papers in Microarchitecture conferences preferred.
• Experiences in developing simulators for high-performance computing systems.
• Understanding of system level characteristics of LLM, DLRM and CNN and other ML workloads.
• Familiarity with PyTorch, TensorFlow, or JAX.
• Strong analytical and problem-solving skills.
• Excellent communication and interpersonal skills.
• Ability to work independently and as part of a team.
• You’re inclusive, adapting your style to the situation and diverse global norms of our people.
• An avid learner, you approach challenges with curiosity and resilience, seeking data to help build understanding.
• You’re collaborative, building relationships, humbly offering support and openly welcoming approaches.
• Innovative and creative, you proactively explore new ideas and adapt quickly to change.

Benefits

• Relocation Support for those living 50+ miles away from the office.
• Housing Stipend for summer internships.
• Charitable giving match and community involvement opportunities.
• Emotional wellness support with on-demand apps and paid therapy sessions.
• Onsite Café and gym, plus virtual classes for fitness.
• Flexible environment to balance work and personal life.