MTS - Machine Learning

Micron Technology
Singapore
Workplace: OnsiteFull timeFunction: Data Science & Machine LearningSkills: ["Analytical thinking","Interpersonal communication","Oral and written communication","Ability to prioritize","Ability to meet timelines"]

Build and deploy scalable AI/ML and GenAI solutions for Micron’s Smart Manufacturing and AI team. You will architect large-scale model training (SFT/RLHF) on multi-node, multi-GPU clusters, optimize GPU performance, and design agentic systems that automate manufacturing workflows. Responsibilities include building data/solution pipelines, optimizing data structures across Snowflake and Google Cloud, and creating CI/CD pipelines for cloud-based ML and AI agent deployments.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
Micron Technology
Micron Technology
2 weeks ago

MTS - Machine Learning

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 13 hours agoStatus: Live

Job Summary

Build and deploy scalable AI/ML and GenAI solutions for Micron’s Smart Manufacturing and AI team. You will architect large-scale model training (SFT/RLHF) on multi-node, multi-GPU clusters, optimize GPU performance, and design agentic systems that automate manufacturing workflows. Responsibilities include building data/solution pipelines, optimizing data structures across Snowflake and Google Cloud, and creating CI/CD pipelines for cloud-based ML and AI agent deployments.
Location: Singapore
Workplace: Onsite
Employment Type: Full time
Job Function: Data Science & Machine Learning
Seniority: Mid level

Key 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 and mixed-precision techniques.
  • •Design and develop autonomous AI Agents for multi-step reasoning, planning, and tool execution to automate manufacturing workflows.
  • •Build and maintain data/solution pipelines and optimize data structures in data management systems to enable AI/ML and agentic solutions.
  • •Create and maintain CI/CD pipelines for ML and AI Agent solutions in the cloud, and profile/debug GPU performance bottlenecks.

Key Requirements

  • •Technical degree required; Computer Science or Statistics background highly desired.
  • •Deep understanding of GPU architecture and managing GPU resources across cloud and on-prem environments.
  • •Hands-on experience with DDP, FSDP, and model parallelism techniques for distributed training.
  • •Proficiency fine-tuning LLMs using PEFT (LoRA/QLoRA) and optimizing inference engines (vLLM, TensorRT-LLM).
  • •Experience building end-to-end ML systems and GenAI/AI Agents using frameworks such as LangChain, LangGraph, LlamaIndex, or AutoGen.
Experience:Machine learningGenAIAI agentsCloudBig dataDistributed systemsDistributed trainingETLManufacturing AI
Skills:Analytical thinkingInterpersonal communicationOral and written communicationAbility to prioritizeAbility to meet timelines
Tech Stack:GPU architectureNVLinkDistributed Data Parallel (DDP)Fully Sharded Data Parallel (FSDP)Model parallelismSFTRLHFFSDPDeepSpeedMegatron-LMFP16BF16LangChainLangGraphCrewAIInternal APIsDatabasesTool executionNsight SystemsPyTorch Profiler

Company Brief

Micron Technology
Designs and manufactures semiconductor memory and storage solutions, including DRAM, NAND, and NOR flash, for computing, networking, mobile, and automotive markets worldwide.
Industry: Electronics Manufacturing
Company Size: Enterprise (1,001+ employees)
Revenue: USD 1B+
Growth: Public Company
Valuation: Public Company (Market Cap in USD)
Funding: IPO / Publicly Listed
Headquarters: Boise, United States
Founded: 1978
Glassdoor
Glassdoor: 3.7
WebsiteLinkedIn