About this role
Responsibilities include architecting and executing large-scale custom model training and fine-tuning jobs on multi-node, multi-GPU clusters, while optimizing throughput and memory efficiency using distributed strategies. The role also involves designing and developing autonomous AI Agents for complex manufacturing workflow automation and analyzing workloads to identify performance bottlenecks.
Candidates must possess deep understanding of GPU architecture and hands-on experience with distributed data parallel techniques, along with proficiency in fine-tuning Large Language Models using PEFT methods. A minimum of 5 years of experience in performance optimization or parallel computing, deep expertise in C++, and strong Python scripting skills are required.
What they're looking for
GPU ArchitecturePerformance OptimizationDistributed TrainingFSDPDeepSpeedMegatron-LMMixed-PrecisionCUDA
Frequently asked questions
What does a MTS, AI Engineering, SMAI at Micron Technology do?
Responsibilities include architecting and executing large-scale custom model training and fine-tuning jobs on multi-node, multi-GPU clusters, while optimizing throughput and memory efficiency using distributed strategies. The role also involves designing and developing autonomous AI Agents for compl…
What skills does this MTS, AI Engineering, SMAI role need?
Key skills for this role include GPU Architecture, Performance Optimization, Distributed Training, FSDP, DeepSpeed, Megatron-LM.
How much does a MTS, AI Engineering, 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 MTS, AI Engineering, SMAI role remote, hybrid, or on-site?
This role is on-site, based in Singapore.
How do I apply for this MTS, AI Engineering, 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.