AI Memory Performance & Power Modeling Intern

Micron Technology, Inc.; Austin (TX); On-site; USD 69,000 - 77,000; Full time

Job summary

Micron Technology, Inc. invites applications for an AI/ML Workload Analysis Intern in Austin, TX. You will partner with system architects and performance engineers to analyze AI workloads, develop memory characterization methodologies, and generate workload traces for training and inference scenarios.

This role focuses on memory behavior, bandwidth needs, and scalability, leveraging simulation, profiling, and data analysis tools to inform future memory technologies and AI platforms.

Qualifications

• Pursuing an MS or PhD in Computer Engineering, Computer Science, Electrical Engineering, Data Science, or a related field.
• Coursework or research experience in: Computer Architecture; Machine Learning / Deep Learning; Parallel Computing; Operating Systems; Data Structures and Algorithms; Performance Analysis.
• Programming experience in: Python; C/C++.
• Strong analytical, problem-solving, and quantitative reasoning skills.
• Excellent written and verbal communication skills.

Responsibilities

• Analyze AI/ML workloads including Large Language Models (LLMs), recommendation models, computer vision, multimodal, and generative AI applications.
• Collect, process, and analyze memory traces from AI training and inference workloads.
• Characterize memory access patterns, locality, reuse behavior, bandwidth utilization, and latency sensitivity.
• Characterize compute, memory, storage, and interconnect behavior across AI training and inference workloads.
• Develop workload traces, performance models, and architecture simulation environments.
• Investigate system bottlenecks related to memory capacity, bandwidth, latency, cache utilization, and communication overheads.
• Evaluate architectural tradeoffs across CPUs, GPUs, NPUs, accelerators, and memory technologies.
• Create automation frameworks for workload execution, data collection, and results analysis.
• Validate simulation and analytical models using benchmark and hardware measurements.
• Present findings through technical reports, visualizations, and architecture reviews.

Skills

Python

C/C++

Machine Learning

Parallel Computing

Operating Systems

Education

MS or PhD in Computer Engineering, Computer Science, Electrical Engineering, Data Science, or related field

Tools

Gem5

Accel-Sim

Nsight

VTune

perf

Job description

Micron Technology, Inc. invites applications for an AI/ML Workload Analysis Intern in Austin, TX. You will partner with system architects and performance engineers to analyze AI workloads, develop memory characterization methodologies, and generate workload traces for training and inference scenarios.

This role focuses on memory behavior, bandwidth needs, and scalability, leveraging simulation, profiling, and data analysis tools to inform future memory technologies and AI platforms.