Principal Engineer, Mle, Smai

Micron Technology

Not specified; not specified; benefits include med...
Not specified
Deep understanding of gpu architecture
Distributed training strategies fsdp deepspeed
Fine-tuning llms using peft techniques
Micron Technology is seeking a Principal Machine Learning Engineer for their Smart Manufacturing and AI team, focusing on the development of advanced AI solutions that enhance memory manufacturing processes. The ideal candidate should possess extensive experience in machine learning, particularly with large-scale model training and AI frameworks, alongside a strong background in GPU architecture and cloud environments

Job Summary

  • The Smart Manufacturing and AI team builds advanced machine learning, GenAI, and agentic AI solutions that directly impact how memory is designed and manufactured.
  • As a Principal Machine Learning Engineer, you will play a critical role in designing and deploying scalable AI systems that power Micron's manufacturing operations.
  • Micron offers a choice of medical, dental and vision plans, paid family leave, and a robust paid time-off program to support employee wellbeing.

Matching Summary

Match Score: 85

Micron Technology is seeking a Principal Machine Learning Engineer for their Smart Manufacturing and AI team, focusing on the development of advanced AI solutions that enhance memory manufacturing processes. The ideal candidate should possess extensive experience in machine learning, particularly with large-scale model training and AI frameworks, alongside a strong background in GPU architecture and cloud environments.

Salary

Not specified; Not specified; Benefits include medical, dental, vision, paid time-off, and paid holidays

Skills & Requirements

Must-have

  • Deep understanding of GPU architecture
  • Distributed training strategies FSDP DeepSpeed
  • Fine-tuning LLMs using PEFT techniques
  • Developing GenAI applications with LangChain
  • Proficiency in Python programming
  • Building end-to-end ML systems

Nice-to-have

  • Experience with HPC job schedulers Slurm
  • Knowledge of CUDA programming and Triton kernels
  • Experience orchestrating GPU workloads on Kubernetes
  • Deep knowledge of math probability statistics
  • Demonstrated ability to transform prototypes to production

Key Requirements

  • 9+ years building scalable ETL pipelines
  • Bachelor's degree in Computer Science or Statistics
  • Strong scripting skills in Python or Java

Work Rights

Not specified

Tailored Resume

Cover Letter