Senior Machine Learning and AI Engineer

Senior/Staff Engineer, Technology -AMAIS

Our vision is to transform how the world uses information to enrich life for all.

Micron Technology is a world leader in innovating memory and storage solutions that accelerate the transformation of information into intelligence, inspiring the world to learn, communicate and advance faster than ever.

Responsibilities include, but not limited to:

• Design, develop, and deploy machine learning models for predictive, prescriptive, and generative tasks using advanced statistical, ML, Deep Learning and analytical techniques.
• Architect and lead the end-to-end development of ML & Agentic workflows from concept to production.
• Build and operationalize agentic workflows capable of autonomous decision-making and adaptive learning, using frameworks like LangGraph, LangChain, CrewAI, and AutoGen.
• Define and implement production-grade agentic architectures with modularity, observability, and fault tolerance to ensure reliability and maintainability.
• Proven hands-on experience with cloud platforms, particularly Google Cloud Platform (GCP), is highly advantageous. Familiarity with cloud-native tools and infrastructure for model training, orchestration, and monitoring will significantly enhance your effectiveness and impact in this role.
• Establish and promote Best Known Methods (BKMs) for deploying ML & agentic solutions, including prompt management, memory handling, and agent orchestration strategies, Error handling and auto-updating of models.
• Continuously evaluate and fine-tune model and agent performance using structured evaluation frameworks, RLHF, prompt tuning, and feedback loops.
• Extract, cleanse, and analyze data from diverse sources using SQL and other query languages, applying techniques for outlier detection and missing data handling.
• Collaborate with cross-functional teams to integrate ML and agentic solutions into scalable systems aligned with business goals and user needs.
• Mentor junior engineers, conduct code reviews, and contribute to technical design and architectural decisions.
• Lead the selection and implementation of ML and agentic tools, frameworks, and infrastructure for efficient development and deployment.
• Stay current with emerging trends in machine learning, agentic AI, and related technologies to drive innovation and continuous improvement.
• Strong software development skills.
• Strong verbal and written communication skills.

Core Technical Skills:

• Programming Languages: Python (preferred), TensorFlow, PyTorch, Scikit-learn, JAX
• Agentic AI Frameworks: LangChain, LangGraph, AutoGen, CrewAI, Google ADK
• Machine Learning Techniques: Supervised, unsupervised, reinforcement learning, deep learning, generative models
• Data Engineering & Analysis: SQL, PySpark, SparkR, SparklyR; outlier detection, missing data handling
• Visualization & Reporting: Tableau or similar tools
• Cloud Platforms: Experience with Google Cloud Platform (GCP) or other cloud environments for scalable ML deployment.
• ML Ops & Infrastructure: MLFlow, Ray, Hugging Face, OpenAI Gym
• Evaluation & Optimization: RLHF, prompt tuning, structured evaluation frameworks

Experience Level:

• 3-7 years of experience in data science, machine learning, or AI engineering roles
• Proven track record of working on ML projects end to end and have experience in deploying ML models and agentic workflows in production
• Prior contributions to technical design, architecture decisions, and mentoring junior engineers
• Publications in top-tier conferences (CVPR, NIPS, ICML, KDD) are a plus, though not required

Soft Skills:

• Strong analytical thinking and problem-solving abilities
• Excellent communication and collaboration skills

About Micron Technology, Inc.

We are an industry leader in innovative memory and storage solutions transforming how the world uses information to enrich life for all. With a relentless focus on our customers, technology leadership, and manufacturing and operational excellence, Micron delivers a rich portfolio of high-performance DRAM, NAND, and NOR memory and storage products through our Micron® and Crucial® brands. Every day, the innovations that our people create fuel the data economy, enabling advances in artificial intelligence and 5G applications that unleash opportunities — from the data center to the intelligent edge and across the client and mobile user experience.

To learn more, please visit micron.com/careers

All qualified applicants will receive consideration for employment without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, veteran or disability status.

To request assistance with the application process and/or for reasonable accommodations, please contact hrsupport_india@micron.com

Micron Prohibits the use of child labor and complies with all applicable laws, rules, regulations, and other international and industry labor standards.

Micron does not charge candidates any recruitment fees or unlawfully collect any other payment from candidates as consideration for their employment with Micron.

AI alert: Candidates are encouraged to use AI tools to enhance their resume and/or application materials. However, all information provided must be accurate and reflect the candidate's true skills and experiences. Misuse of AI to fabricate or misrepresent qualifications will result in immediate disqualification.   

Fraud alert: Micron advises job seekers to be cautious of unsolicited job offers and to verify the authenticity of any communication claiming to be from Micron by checking the official Micron careers website in the About Micron Technology, Inc.

Experience Level

Senior Level