HBM Memory Architect for Generative AI Workloads

Micron Technology, Inc.; Folsom (CA); On-site; USD 103,000 - 210,000; Full time

Benefits offered by this job

Medical, dental and vision plans

Paid time-off

Paid holidays

Job summary

Micron Technology, Inc. is seeking a Memory System Architect to define future memory architectures for AI accelerators and data-intensive workloads.

You will research, model, and evaluate architectures supporting Generative AI, with cross-functional collaboration to guide product roadmaps. The role requires expertise in memory systems, HBM/DRAM, and performance optimization, along with strong programming skills and communication abilities.

Qualifications

• Master's degree or PhD in Computer Architecture, Electrical Engineering, Computer Engineering, Computer Science, or a related technical field.
• Strong understanding of GPU, TPU, AI accelerator, or large-scale computing architectures.
• Experience with performance modeling, architectural simulation, system analysis, or related research methodologies.

Responsibilities

• Research and evaluate advanced memory architecture concepts for AI accelerators, including GPU, TPU, and emerging compute platforms.
• Develop and analyze system-level models to assess performance, power, scalability, and efficiency across memory and compute architectures.
• Investigate memory hierarchy optimizations involving HBM, DRAM, cache, storage, and interconnect technologies.
• Evaluate architectural trade-offs and emerging technologies, including compute-near-memory concepts, to support future product development.
• Collaborate with cross-functional teams to translate research findings into architecture recommendations and long-term technology roadmaps.

Skills

Memory architectures

GPU/TPU/AI accelerator

Memory hierarchy

Performance modeling

Python

C++

System analysis

Communication & collaboration

Education

Master's degree or PhD in Computer Architecture/EE/CE/CS

Tools

CUDA

TensorFlow

Job description

Micron Technology, Inc. is seeking a Memory System Architect to define future memory architectures for AI accelerators and data-intensive workloads.

You will research, model, and evaluate architectures supporting Generative AI, with cross-functional collaboration to guide product roadmaps. The role requires expertise in memory systems, HBM/DRAM, and performance optimization, along with strong programming skills and communication abilities.