M

Micron Technology

Makes memory and storage chips for AI and PCs

Principal AI Systems Performance Engineer

AI & Machine LearningPosted on Nov 2
Full-timeAustin, TX, USANo salary listedIn-person

About the Job

Requirements

  • Bachelor’s degree with at least 10 years of experience; master’s degree with 5 or more years; or Ph.D. with demonstrated experience in a related field.
  • Hands-on experience profiling and benchmarking multi-GPU AI systems across single-node and multi-node environments.
  • Understanding of transformer workloads, memory behavior, distributed systems, and agentic AI infrastructure.
  • Experience with PyTorch, an AI serving platform, Python, and at least one of C, C++, CUDA, or Triton.
  • Ability to lead controlled experiments and communicate results to technical and business audiences.

Responsibilities

  • Develop and perform controlled, representative AI experiments on current and pre-production CPU-GPU systems, evaluating memory and storage performance, power, and thermal tradeoffs.
  • Profile workloads from GPU memory through system memory, CXL-attached memory, storage, and interconnects; assess performance from node to cluster scale, including efficiency, latency, utilization, scalability, and tail behavior.
  • Evaluate data placement, migration, prefetching, compression, and tiering approaches for KV cache, model state, and other AI data.
  • Build and validate performance models, simulations, traces, and reusable benchmarking tools to forecast future system requirements.
  • Lead cross-functional projects and transform outcomes into product requirements, develop recommendations, publications, demonstrations, and customer mentorship.

Desired Qualifications

  • Experience with memory tiering, KV-cache management, or emerging AI workloads.
  • Experience with disaggregated infrastructure or performance modeling and simulation.
  • Relevant research, patents, open-source contributions, or other substantial technical work.

About the Company

Micron Technology

Makes memory and storage chips for AI and PCs

Micron Technology designs and manufactures memory and storage chips, mainly DRAM and NAND flash, used in data centers, PCs, phones, cars and industrial equipment. It is one of only three companies that make HBM, the stacked high-bandwidth memory placed beside processors in AI servers, and it is the largest memory chipmaker based in the United States. Unlike chip designers that outsource production, Micron runs its own fabs and assembly sites in the United States and across Asia. It earns money by selling chips, memory modules and solid-state drives to cloud providers, device makers and distributors. Micron was founded in 1978 and is headquartered in Boise, Idaho.

Company size
10,001+
Company stage
IPO
Headquarters
Boise, ID
Founded
1978

Recently Posted Jobs

Explore more opportunities at Micron Technology

See more jobs