HBM Memory Architect for Generative AI Compute

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

Benefits offered by this job

Medical Insurance

Dental Insurance

Vision Insurance

Paid Time Off

Paid Holidays

Equity

Job summary

Micron Technology is seeking a Memory System Architect for the HBM Architecture team to shape future memory architectures supporting Generative AI and data-heavy workloads. You will research memory system concepts, analyze AI accelerator architectures, and build models guiding product strategy and roadmaps for next-generation memory solutions.

The role involves evaluating memory hierarchy optimizations, collaborating with cross-functional teams, and contributing to architecture recommendations

Qualifications

• 0-2 years of relevant experience in computer architecture, memory systems, AI hardware, or HPC.
• Master's degree, PhD, or equivalent experience in Computer Architecture, Electrical Engineering, Computer Engineering, CS, or related field.
• Strong understanding of GPU, TPU, AI accelerator, or large-scale computing architectures.
• Knowledge of memory hierarchy concepts including caches, HBM, DRAM, storage systems, and memory performance optimization.
• Experience with performance modeling, architectural simulation, system analysis, or related research methodologies.
• Programming experience in Python, C++, and/or AI computing frameworks such as CUDA, TensorFlow, or similar technologies.
• Strong analytical, problem-solving, communication, and collaboration skills.
• Familiarity with AI-enabled tools and large language models (LLMs) such as Copilot, Claude, or similar technologies.

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

GPU/TPU/AI accelerator architectures

Performance modeling

Python

C++

CUDA/TensorFlow

Education

Master's degree or PhD in Computer Architecture or related field

Tools

CUDA

TensorFlow

Job description

Micron Technology is seeking a Memory System Architect for the HBM Architecture team to shape future memory architectures supporting Generative AI and data-heavy workloads. You will research memory system concepts, analyze AI accelerator architectures, and build models guiding product strategy and roadmaps for next-generation memory solutions.

The role involves evaluating memory hierarchy optimizations, collaborating with cross-functional teams, and contributing to architecture recommendations