Senior/Staff Engineer, Technology - AMAIS

Micron Technology; Hyderabad; On-site; INR 1,200,000 - 1,800,000; Full time

Job summary

Micron Technology, located in Hyderabad, is seeking a Senior/Staff Data Science Engineer to design and deploy machine learning models and develop agentic workflows. The role involves working with cross-functional teams to integrate ML solutions while mentoring junior engineers. Candidates should have 3 to 7 years of experience in data science or AI, strong skills in Python, and familiarity with cloud platforms like GCP. Micron promotes a diverse and inclusive workplace.

Qualifications

• 3–7 years of experience in data science or AI engineering.
• Proven experience with ML projects end-to-end.
• Strong software development skills.

Responsibilities

• Design, develop, and deploy machine learning models.
• Build and operationalize agentic workflows.
• Collaborate with cross-functional teams to integrate solutions.

Skills

Machine Learning

Python

Statistical Analysis

Data Engineering

Cloud Computing

SQL

Tools

Google Cloud Platform

TensorFlow

PyTorch

Tableau

Job description

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.

As a Senior/ Staff Data Science Engineer at Micron, you will employ techniques and theories drawn from areas of mathematics, statistics, semiconductor physics, materials science, and information technology to uncover patterns in data from which predictive models, actionable insights, and solutions can be developed.

You will work with expert Data Scientists, Data Engineers, Business Areas Engineers, and UX teams to identify questions and issues for data analysis projects and improvement of existing tools. In this position, you will develop software programs, algorithms, and automated processes to cleanse, integrate, and evaluate large datasets from multiple disparate sources. There will be significant opportunities to perform exploratory and new solution development activities.

Responsibilities

• 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 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

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 [e-mail].