Systems Performance Engineer

3M HEALTHCARE; Austin (TX); On-site; USD 150,000 - 210,000; Full time

Benefits offered by this job

Medical, dental, and vision coverage

Income protection benefits

Paid family leave

Paid time off and holidays

Job summary

Micron Technology is seeking an engineer to work on AI training and inference systems, focusing on LLM execution engines, memory hierarchies, and performance optimization across data-center platforms.

The role involves end‑to‑end profiling, benchmarking, and collaboration with senior engineers and researchers. You will develop tools and methods to improve throughput, latency, and resource utilization for large-scale AI workloads.

Qualifications

• Bachelor’s or Master’s degree in CS/EE or equivalent experience.
• Proficiency in C/C++ and Python.
• Experience in Linux environments, debugging, profiling, and automation.
• Solid understanding of memory systems, NUMA, GPUs, and server architectures.
• Strong written and verbal communication skills.

Responsibilities

• Profile and optimize LLM training and inference workloads.
• Design and evaluate KV-cache and state-management strategies for LLM serving.
• Build benchmarking, simulation, and emulation frameworks for AI workloads across memory tiers.
• Develop data placement, migration, and prefetching algorithms for heterogeneous memory pools.
• Characterize LLM execution engines and analyze throughput and latency across deployments.
• Collaborate with engineering, architecture, and research teams to influence future AI systems.

Skills

C/C++ programming

Python

Linux

Memory systems

Performance profiling

System optimization

Communication

Education

Bachelor’s or Master’s in CS/EE

Tools

gem5

Ramulator

Perf tooling

Job description

Our vision is to transform how the world uses information to enrich life for all. 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.

The engineer works with senior engineers and researchers on AI training and inference systems, focusing on LLM execution engines, data and KV‑cache management, and multi‑tier memory hierarchies across modern data‑center platforms. The role centers on end‑to‑end performance characterization and optimization of large‑scale AI workloads, spanning single‑node GPUs to rack‑scale inference deployments.

Key Responsibilities

• Build, develop, and improve systems software tools for profiling, tracing, and analyzing LLM training and inference workloads.
• Design and evaluate KV‑cache and state‑management strategies for LLM serving, including reuse, eviction, compression, tiering, and lifecycle management.
• Build and extend benchmarking, simulation, and emulation frameworks for AI inference and training across heterogeneous memory tiers.
• Develop and evaluate data placement, migration, and prefetching algorithms across HBM, LP/DRAM, CXL memory pools, NVMe, and remote memory systems.
• Characterize and optimize LLM execution engines (prefill/decode), including attention behavior, batching strategies, and token‑level performance.
• Analyze rack‑scale and cluster‑scale inference deployments, focusing on throughput, latency, utilization, cost, and token economics.
• Develop workloads that reflect real customer AI systems, including LLM serving, agentic pipelines, retrieval‑augmented generation, multimodal inference, and long‑context workloads.
• Instrument and analyze performance across GPUs, CPUs, memory subsystems, interconnects, and storage, identifying end‑to‑end bottlenecks.
• Evaluate system interactions across OS, runtime layers, containerized deployments, and distributed inference stacks.
• Automate performance measurement, experimentation, and analysis workflows to improve repeatability and scale.
• Summarize findings into clear methodologies, internal reports, and technical presentations for engineering and leadership audiences.
• Collaborate across engineering, architecture, and research teams, and with external academic and industry partners.
• Provide actionable feedback to product, architecture, and platform teams to influence future AI systems and memory designs.

Required Qualifications

• Bachelor’s or Master’s degree, or equivalent experience, in Computer Science, Electrical Engineering, or a related field.
• Strong foundation in operating systems, memory systems, parallel computing, or distributed systems.
• Proficiency in systems programming and analysis using C/C++ and Python.
• Experience working in Linux environments, including debugging, profiling, and automation.
• Solid understanding of modern server architectures, including GPUs, CPUs, cache hierarchies, NUMA, and memory subsystems.
• Experience analyzing performance data and reasoning about system‑level behavior.
• Strong written and verbal communication skills.
• Ability to work independently on scoped problems and collaboratively on larger system efforts.

Preferred Qualifications

• Experience with LLM training and inference systems, including execution runtimes and serving frameworks.
• Hands‑on experience with KV cache management, long‑context execution, or stateful inference workloads.
• Familiarity with GPU architectures and AI accelerators, including memory and interconnect behavior.
• Experience with multi‑tier memory systems, including HBM, LP/DRAM, CXL‑attached memory, NVMe, and remote/disaggregated memory.
• Experience profiling and optimizing AI inference pipelines, including batching, scheduling, and latency‑sensitive workloads.
• Familiarity with agentic AI frameworks, multi‑agent systems, or workflow‑based inference pipelines.
• Experience with distributed AI systems, rack‑scale deployments, or cluster‑level performance analysis.
• Exposure to memory or system simulators (e.g., gem5, Ramulator) or analytical performance modeling.
• Familiarity with containers, orchestration, and AI infrastructure stacks.
• Experience applying machine learning techniques to systems optimization or performance analysis.

Benefits

Micron offers comprehensive medical, dental, and vision coverage, income protection benefits, paid family leave, paid time off, and paid holidays. Additional details are available in Micron’s Benefits Guide.

Equal Employment Opportunity Statement

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.