Member of Technical Staff (MTS), Machine Learning, SMAI
Core
Architecting and executing large-scale custom model training and fine-tuning jobs, optimizing GPU performance, and building autonomous AI Agents to automate complex manufacturing workflows for Micron's memory solutions.
Role type
Senior IC Machine Learning Engineer (Member of Technical Staff)
Builds
Scalable AI/ML solutions, GenAI applications, and autonomous AI Agents for manufacturing processes.
Domain
Semiconductor manufacturing / Memory solutions / Generative AI
Deliverable
production ML models
Required skills
GPU architecture management, Distributed Data Parallel (DDP), Fully Sharded Data Parallel (FSDP), Large Language Model fine-tuning (PEFT, LoRA, QLoRA), Inference engine optimization (vLLM, TensorRT-LLM), GenAI application development, LLM prompt engineering and function calling, End-to-end ML system automation, PyTorch, Python, CI/CD pipelines, Big data processing, ETL pipeline construction
Preferred skills
HPC job schedulers (Slurm), GPU workload orchestration on Kubernetes (Ray, KubeFlow), Lower-level optimization (CUDA, Triton kernels, C++ extensions), Multi-Agent Systems, Advanced mathematics and statistics, Computer vision and signal processing
Technologies
PyTorch, LangChain, LangGraph, CrewAI, LlamaIndex, AutoGen, Snowflake, Google Cloud Platform, Jenkins, Git, Docker, Kubernetes, Nsight Systems, PyTorch Profiler, Slurm, Ray, KubeFlow, CUDA, Triton
Responsibilities
Architect and execute large-scale custom model training and fine-tuning jobs on multi-node, multi-GPU clusters; Optimize training throughput and memory efficiency using distributed training strategies; Design and develop autonomous AI Agents capable of multi-step reasoning and tool execution; Implement Agentic frameworks to orchestrate LLM interactions with internal APIs and databases; Profile and debug GPU performance bottlenecks to maximize hardware utilization; Build and maintain data/solution pipelines that feed machine learning models and GenAI applications; Design and optimize data structures in data management systems to enable AI/ML solutions; Create and maintain CI/CD pipelines for machine learning and AI Agent solutions in the cloud.
Seniority
Senior, hands-on IC