Staff Machine Learning Engineer

We are seeking a Staff Machine Learning Engineer to join our team at Micron Technology. The successful candidate will analyze data sets, design and implement machine learning models, and build knowledge in machine learning and AI. The role will also involve building and maintaining data infrastructure, collaborating with product and engineering teams, and implementing strategies for deploying machine learning models.

Requirements

• Master's degree in Computer Science, Machine Learning, Data Science, Statistics, or related field
• 3 years of experience in the job offered or in a related occupation
• Building and executing end to end ML systems
• TensorFlow, PyTorch, and scikit learn
• Prediction (regression) and classification algorithms; deep learning, reinforcement learning, and generative AI
• Python or Java
• Pub/Sub, Solace, or Kafka
• Developing ML/AI solutions in GCP, Azure, or AWS
• Docker and Kubernetes
• Developing ETL/ELT pipelines using Kubeflow, Dataflow, or Airflow
• SQL
• Data structures and schemas
• APIs to expose trained machine learning models for consumption
• Building enterprise-grade applications using Generative AI technologies
• Retrieval-Augmented Generation (RAG) for knowledge grounding and fine-tuning pre-trained models on domain-specific datasets
• Designing and implementing agentic systems that orchestrate autonomous workflows and integrate with APIs

Benefits

• Choice of medical, dental, and vision plans
• Paid family leave
• Paid time-off program
• Paid holidays
• Benefit programs to protect income in case of illness or injury