AI Systems Performance Architect

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

Job summary

Micron Technology, Inc. is seeking a Principal AI Systems Performance Engineer to direct experiments on AI training, inference, and agentic workloads across server systems.

You will identify bottlenecks in bandwidth, capacity, latency, power, and data movement, then translate findings into product requirements and guidance for engineering, product, and customer teams. You will profile and benchmark multi-GPU AI systems, develop performance models, and lead cross-functional projects to shape

Qualifications

• Bachelor’s degree with at least 10 years of experience; master’s with 5+ years; or Ph.D. with demonstrated related experience.
• 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 AI experiments on current and pre-production CPU-GPU systems, evaluating memory, 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.
• Evaluate data placement, migration, prefetching, compression, and tiering for AI data such as KV cache and model state.
• Build and validate performance models, simulations, traces, and benchmarking tools to forecast future system requirements.
• Lead cross-functional projects and transform outcomes into product requirements and customer guidance.

Skills

Profiling multi-GPU systems

Transformer workloads

PyTorch

Python

C/C++/CUDA/Triton

Memory tiering

KV-cache management

Disaggregated infrastructure

Patents/open-source work

Education

Bachelor’s degree + 10+ years

Master’s degree + 5+ years

Ph.D. in related field

Tools

CUDA

Triton

Job description

Micron Technology, Inc. is seeking a Principal AI Systems Performance Engineer to direct experiments on AI training, inference, and agentic workloads across server systems.

You will identify bottlenecks in bandwidth, capacity, latency, power, and data movement, then translate findings into product requirements and guidance for engineering, product, and customer teams. You will profile and benchmark multi-GPU AI systems, develop performance models, and lead cross-functional projects to shape