PhD Intern: AI Memory & Architecture for Accelerators

Micron Technology; Folsom (CA); On-site; USD 143,270,000 - 144,648,000; Full time

Benefits offered by this job

Health benefits

Paid time off

Paid holidays

Equal opportunity employer

Job summary

Micron Technology is seeking a PhD Intern on the HBM Architecture team to research and prototype memory and system architecture concepts for next-generation AI accelerators. You will develop performance models, analyze AI workloads, and evaluate how memory hierarchy and interconnect choices impact system performance and power efficiency.

You will collaborate with experienced architects and engineers, exploring AI/LLM-based tools to enhance research workflows and contribute to future AI computing

Qualifications

• Currently pursuing a PhD in computer architecture or related field.
• 0-2 years of relevant research or internship experience in computer architecture, memory systems, or AI hardware.
• Knowledge of memory system architecture, including caches, HBM, DRAM, memory hierarchy, and interconnect concepts.
• Experience with performance modeling, architectural simulation, system analysis, or related research methodologies.
• Programming in Python, C++, or similar languages.
• Strong analytical, problem-solving, and technical communication skills.

Responsibilities

• Research and evaluate advanced memory and system architecture concepts for AI accelerators, including GPUs, TPUs, and custom accelerator designs.
• Develop and analyze performance models to assess bandwidth, latency, power, scalability, and overall system efficiency.
• Analyze AI training, inference, and HPC workloads to identify memory and system performance bottlenecks.
• Investigate memory hierarchy, HBM, DRAM, cache, storage, and interconnect optimizations to improve system performance.
• Collaborate with cross-functional engineering teams and explore use of AI/LLM-based tools to enhance research workflows.

Skills

Python

C++

Analytical skills

Communication

Education

PhD in Computer Architecture or related field

Tools

SystemC

GPU profiling tools

Emulation platforms

Architectural simulators

PyTorch

TensorFlow

CUDA-based environments

Job description

Micron Technology is seeking a PhD Intern on the HBM Architecture team to research and prototype memory and system architecture concepts for next-generation AI accelerators. You will develop performance models, analyze AI workloads, and evaluate how memory hierarchy and interconnect choices impact system performance and power efficiency.

You will collaborate with experienced architects and engineers, exploring AI/LLM-based tools to enhance research workflows and contribute to future AI computing