MTS - Machine Learning

Full-time Other-General

Job Overview

The Smart Manufacturing and AI team at Micron Technology is looking for an ambitious Machine Learning Engineer (Member of Technical Staff | MTS). Our mission is to deliver industry‑winning machine learning, custom GenAI, and Agentic AI solutions to power Micron’s dominance in the highly competitive memory solutions market.

Responsibilities

• Architect and execute large‑scale custom model training and fine‑tuning jobs (SFT, RLHF) on multi‑node, multi‑GPU clusters.
• Optimize training throughput and memory efficiency using distributed training strategies (FSDP, DeepSpeed, Megatron‑LM) and mixed‑precision techniques (FP16/BF16).
• Design and develop autonomous AI Agents capable of multi‑step reasoning, planning, and tool execution to automate complex manufacturing workflows.
• Implement Agentic frameworks (e.g., LangChain, LangGraph, CrewAI) to orchestrate LLM interactions with internal APIs, databases, and software tools.
...