Intern - Memory Performance and Power Modeling at Micron Technology | Austin, TX, US | Rezi

Intern - Memory Performance and Power Modeling at Micron Technology

Intern - Memory Performance and Power Modeling

Micron Technology · Austin, TX, US

1 weeks ago

Intern - Memory Performance and Power Modeling

Micron Technology · Austin, TX, US

10 days ago
Resume preview

Impress employers and recruiters.
Choose from hundreds of resume examples.

Target Resume Now
Resume preview

Tailor your resume to this Intern - Memory Performance and Power Modeling role.

Rezi rewrites your resume against Micron Technology's job description. Free.

Resume score gauge reading 58 out of 100

Don't guess if your resume is good enough.

See how it scores against the Intern - Memory Performance and Power Modeling posting at Micron Technology — free, in seconds.

About the Role

As an AI/ML Workload Analysis Intern, you will collaborate with system architects and performance engineers to analyze emerging AI workloads and evaluate next-generation computing and memory architectures. Your work will involve developing methodologies to characterize memory behavior, generate workload traces, and identify performance bottlenecks in AI training and inference applications, contributing to architectural decisions for future memory technologies and AI platforms.

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.

Requirements

  • 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, and Performance Analysis.
  • Programming experience in Python, C/C++.
  • Strong analytical, problem-solving, and quantitative reasoning skills.
  • Excellent written and verbal communication skills.

Skills

  • Python
  • C/C++
  • PyTorch
  • TensorFlow
  • JAX
  • ONNX
  • LLMs
  • Transformers
  • Diffusion models
  • Recommender systems
  • Vision models
  • Workload profiling
  • Tracing
  • Performance analysis tools
  • Simulation
  • Modeling
  • Architecture research
  • AI code assistant
  • Nsight
  • VTune
  • perf
  • rocprof
  • GPU architectures
  • NPU architectures
  • TPU architectures
  • Accelerator architectures
  • Memory systems
  • Cache hierarchies
  • DRAM
  • Gem5
  • Accel-Sim
  • Linux development environments
  • Data analysis frameworks

Location

  • US

Work Type

  • Full-time

Experience Level

  • Intern

Education Level

  • MS or PhD

Salary/Compensations

  • $50.24 - $55.82 an hour

Benefits

  • Choice of medical, dental and vision plans
  • Benefit programs that help protect your income if you are unable to work due to illness or injury
  • Paid family leave
  • Robust paid time-off program
  • Paid holidays

About the Company

  • Micron Technology is a world leader in innovating memory and storage solutions that accelerate the transformation of information into intelligence, inspiring the world to learn, communicate and advance faster than ever.
  • As a world leader in the semiconductor industry, Micron is dedicated to your personal wellbeing and professional growth.

Equal Opportunity

  • Micron is proud to be an equal opportunity workplace and is an affirmative action employer. All qualified applicants will receive consideration for employment without regard to race, color, religion, sex, sexual orientation, age, national origin, citizenship status, disability, protected veteran status, gender identity or any other factor protected by applicable federal, state, or local laws.
  • Micron Prohibits the use of child labor and complies with all applicable laws, rules, regulations, and other international and industry labor standards.
  • Micron does not charge candidates any recruitment fees or unlawfully collect any other payment from candidates as consideration for their employment with Micron.