Micron Technology logo

Principal Engineer, Machine Learning, SMAI

Micron Technology

SingaporeFull-timeOn-site

Posted 2 Jun 2026

About this role

The role involves architecting and executing large-scale custom model training and fine-tuning jobs on multi-node GPU clusters, while also designing and developing autonomous AI Agents for complex manufacturing workflow automation. Responsibilities include optimizing training throughput, implementing Agentic frameworks, and building/maintaining data/solution pipelines feeding these AI/ML solutions. Candidates must possess deep expertise in GPU architecture, distributed training techniques (DDP, FSDP), and proficiency in fine-tuning LLMs using PEFT methods, along with experience developing GenAI applications and AI Agents using frameworks like LangChain. Minimum qualifications require 9+ years in scalable ETL pipelines and big data processing, strong Python skills, and familiarity with ML frameworks like PyTorch.

What they're looking for

Machine LearningGenAIAgentic AIDistributed TrainingFSDPDeepSpeedMegatron-LMRLHF

Frequently asked questions

What does a Principal Engineer, Machine Learning, SMAI at Micron Technology do?

The role involves architecting and executing large-scale custom model training and fine-tuning jobs on multi-node GPU clusters, while also designing and developing autonomous AI Agents for complex manufacturing workflow automation. Responsibilities include optimizing training throughput, implementin…

What skills does this Principal Engineer, Machine Learning, SMAI role need?

Key skills for this role include Machine Learning, GenAI, Agentic AI, Distributed Training, FSDP, DeepSpeed.

How much does a Principal Engineer, Machine Learning, SMAI at Micron Technology pay?

The employer did not list a salary for this role. Most similar Singapore roles publish their band on the job page.

Is this Principal Engineer, Machine Learning, SMAI role remote, hybrid, or on-site?

This role is on-site, based in Singapore.

How do I apply for this Principal Engineer, Machine Learning, SMAI role?

You can apply directly on Micron Technology's careers page. ApplyLah can tailor your résumé and cover letter to this exact role in seconds first.