Taro Logo

Machine Learning Engineer

World leader in innovating memory and storage solutions that accelerate the transformation of information into intelligence.
Richardson, TX, USA
Machine Learning
Senior Software Engineer
Hybrid
5,000+ Employees
5+ years of experience
AI · Hardware
This job posting may no longer be active. You may be interested in these related jobs instead:

Description For Machine Learning Engineer

Micron Technology, a global leader in memory and storage solutions, is seeking a Machine Learning Engineer to join their Scalable Memory Systems group. This role is part of a pathfinding and strategy organization that shapes the future of memory for high-performance computing (HPC) and artificial intelligence systems. The position offers a unique opportunity to work with cutting-edge technology, including proof-of-concept computer systems with multiple host servers, GPUs, and memory accelerators.

The role combines deep technical expertise in machine learning with systems architecture knowledge, requiring candidates to conduct AI research, analyze use cases, and collaborate on memory innovations. The ideal candidate should have extensive experience with AI/ML frameworks, deep understanding of neural networks, and strong programming skills in Python and C++.

Working in a hybrid environment from Richardson, TX, you'll be part of a dynamic team that includes both on-site and remote members. The position offers comprehensive benefits including medical, dental, and vision coverage, paid time off, and family leave. This is an excellent opportunity for experienced ML engineers who want to impact the future of memory systems in AI applications.

The role requires either a PhD, master's degree with 5+ years of experience, or bachelor's degree with 10+ years of experience, demonstrating Micron's commitment to hiring seasoned professionals. You'll be working at the intersection of AI and hardware, making this an unique opportunity for those interested in both machine learning and computer architecture.

Last updated 5 days ago

Responsibilities For Machine Learning Engineer

  • Researching technologies and tools related to ML and AI
  • Developing and analyzing AI workflows to support use case studies
  • Collaborating with system architects and software engineers to define memory innovations
  • Modifying applications and libraries for technology demonstrations
  • Sharing knowledge through white papers and presentations

Requirements For Machine Learning Engineer

Python
  • PhD, master's degree with ≥5 years' experience, or bachelor's degree with ≥10 years' experience
  • Experience implementing AI workflows using AI/ML frameworks (PyTorch, TensorFlow)
  • Deep technical understanding of ML and AI algorithms, neural networks and models
  • Knowledge of compute system architecture and memory scaling
  • Experience with application performance benchmarking on heterogeneous systems
  • Python and C++ programming experience in a Linux environment
  • Experience programming GPUs (CUDA, OpenCL, SYCL)
  • Familiarity with compute scaling, parallel computing and shared memory technologies

Benefits For Machine Learning Engineer

Medical Insurance
Dental Insurance
Vision Insurance
Parental Leave
  • Choice of medical, dental and vision plans
  • Paid family leave
  • Robust paid time-off program
  • Paid holidays
  • Income protection for illness or injury

Interested in this job?