Staff Engineer, Advanced Modeling & AI Solutions jobs in United States
cer-icon
Apply on Employer Site
company-logo

Micron Technology · 4 months ago

Staff Engineer, Advanced Modeling & AI Solutions

Micron Technology is a world leader in innovating memory and storage solutions. They are seeking a Staff Engineer for Advanced Modeling & AI Solutions, responsible for deploying intelligent modeling solutions and architecting AI systems to enhance semiconductor processes.

ComputerHardwareManufacturingSemiconductor
check
H1B Sponsor Likelynote

Responsibilities

Drive the deployment of intelligent, advanced physics or ML-powered modeling solutions that enhance semiconductor process technology development, package development, design, and early manufacturing workflows—leveraging large-scale, unstructured and often sparse data to deliver robust, domain-specific AI systems from problem definition to production
Architect and implement agentic AI systems that integrate with Advanced Modeling tools, design automation tools / environments, and product/fab/manufacturing test platforms to accelerate digital twin development for fab technology co-optimization covering wafer, die, and package level models across tech nodes
Establish and promote Best Known Methods (BKMs) for deploying LLMs and agentic systems in production environments, ensuring reliability, efficiency, and maintainability
Benchmark and evaluate model performance using structured evaluation frameworks, and continuously refine models through prompt tuning, RLHF, and feedback loops
Accelerate insight generation from predictive modeling, images, and data. Communicate technical insights and solution strategies clearly to both technical and non-technical stakeholders through compelling data storytelling and visualizations

Qualification

PythonAI/ML project managementLLM deploymentCloud platformsML frameworksData processing platformsSemiconductor experienceCommunication skillsCollaboration

Required

Must have a Master's or PhD in Electrical Engineering, Computer Science, or a related field
Proficient in Python, with over 5 years of experience working with a variety of semiconductor design & process datasets
Proficient in leveraging enterprise data platforms such as Snowflake, BigQuery, MSSQL, Oracle, and AWS Redshift for scalable data processing and analysis
Hands-on experience building and managing AI/ML projects involving LLMs, RAG, and agentic workflows deployed in semiconductor environments
Demonstrated expertise in ML frameworks such as PyTorch or TensorFlow
Strong knowledge of domain-adapted LLM training with practical experience, including pretraining, post training on in-domain synthetic datasets, and model quantization for efficient deployment in semiconductor environments
Must have experience working with cloud platforms such as GCP, AWS, or Azure, including deployment of ML pipelines in production environments
Familiarity with LLM evaluation and benchmarking techniques, including RLHF, prompt tuning, and reward modeling for hardware-aware use cases
Proven ability to independently drive AI/ML projects from problem scoping through deployment in semiconductor environments
Excellent communication skills with the ability to collaborate with process technology, silicon design, design verification, validation and product engineering teams, and translate complex AI/ML concepts into actionable solutions for hardware development

Preferred

Experience includes direct application of AI/ML to semiconductor design, processing, verification, validation, and manufacturing workflows
Deep understanding of semiconductor-specific AI/ML applications
Experience with CI/CD pipelines and MLOps practices for ML/LLM deployment
Strong understanding of agentic AI frameworks (e.g., LangGraph, AutoGen) and evaluation tools (e.g., AgentEval)
Demonstrated ability to translate complex technical concepts into actionable insights for cross-functional teams

Benefits

Micron benefits are designed to help you stay well, provide peace of mind and help you prepare for the future.
We offer a choice of medical, dental and vision plans in all locations enabling team members to select the plans that best meet their family healthcare needs and budget.
Micron also provides benefit programs that help protect your income if you are unable to work due to illness or injury, and paid family leave.
Additionally, Micron benefits include a robust paid time-off program and paid holidays.

Company

Micron Technology

company-logo
Micron Technology is a semiconductor company that produces DRAM, SDRAM, flash memory, SSD and CMOS image sensing chips.

H1B Sponsorship

Micron Technology has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (573)
2024 (429)
2023 (299)
2022 (626)
2021 (564)
2020 (538)

Funding

Current Stage
Public Company
Total Funding
$8.9B
Key Investors
U.S. Department of Commerce
2025-04-24Post Ipo Debt· $1.75B
2025-01-16Post Ipo Debt· $1B
2024-04-18Grant· $6.1B

Leadership Team

leader-logo
Sanjay Mehrotra
Chairman, President & CEO
linkedin
leader-logo
Gursharan Singh
Senior Vice President BE Operations
linkedin
Company data provided by crunchbase