Micron Technology logo

Principal AI Systems Performance Engineer

Micron Technology
NewPosted today

LOCATION

Hyderabad · Onsite

EXPERIENCE

Not specified

SKILLS REQUIRED

Performance OptimizationDistributed SystemsGPU ProgrammingPerformance ModelingPytorchModel serving

Job description

Overview

Direct controlled AI workload performance experiments on complete server systems to identify bandwidth, capacity, latency, power and data-movement limits, and convert findings into product requirements and guidance for engineering, product and customer teams.

What you'll do

  • 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.

What you'll need

  • 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.

Nice to have

  • 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.

Details

  • Location: Hyderabad, Telangana, India.

Read the full description and apply on the company’s own careers page.

Stay safe

Hiring on Abekus is free for applicants

We never charge a fee, and employers are prohibited from doing so. If a recruiter asks for payment, please report them right away.

Principal AI Systems Performance Engineer