Staff AI Engineer — GPU Performance & GenAI

Micron Technology, Inc; Boise (ID); On-site; USD 120,000 - 150,000; Full time

Benefits offered by this job

Medical, dental, and vision plans

Paid time off and holidays

Income protection programs

Job summary

Micron Technology, Inc in Boise, Idaho is looking for a GPU Performance Engineer to drive large-scale AI systems on GPU platforms. This role involves optimizing training workflows and collaborating with cross-functional teams.

The ideal candidate should possess a strong background in GPU architecture, performance optimization, and have hands-on experience with deep learning frameworks. Micron offers comprehensive benefits and a supportive work environment.

Qualifications

• 10+ years of experience in GPU architecture and resource management.
• 5+ years in performance optimization and parallel computing.
• Deep proficiency in LLMs and AI systems.

Responsibilities

• Architect and complete large-scale model training on multi-GPU clusters.
• Analyze and profile workloads to identify compute bottlenecks.
• Collaborate with Hardware Architects for next-gen GPUs.

Skills

GPU architecture

Performance optimization

Parallel computing

C++

Deep learning

Python

Education

Bachelor's or Master’s degree in Computer Science, Statistics, or a related field

Tools

CUDA

PyTorch

Docker

Kubernetes

Job description

Micron Technology, Inc in Boise, Idaho is looking for a GPU Performance Engineer to drive large-scale AI systems on GPU platforms. This role involves optimizing training workflows and collaborating with cross-functional teams.

The ideal candidate should possess a strong background in GPU architecture, performance optimization, and have hands-on experience with deep learning frameworks. Micron offers comprehensive benefits and a supportive work environment.