Micron

Staff Machine Learning Engineer

Posted on: 22 Jun 2024

Bay Area, CA

Job Description

About the Team

The AI / Machine Learning team is building the industry's leading Machine Learning models for use cases at scale. This team is also responsible for pushing the boundaries of applied Machine Learning and state-of-the-art Generative AI techniques on a challenging and diverse dataset. The team is at the forefront of building frameworks for Responsible AI development.

Primary Responsibilities:

In this senior-level role, you will own, train, build and deploy cutting edge deep learning models across all Eightfold products, end to end.
Build on top of Open Source LLM (Large Language Models) to leverage a diverse dataset. 
Apply innovative solutions from Generative AI
Create industry best practices for Machine Learning for Recruiting and HR industry around the globe
Do it responsibly to provide equal opportunity for everyone by extending our internal model fairness platform
Create innovative algorithms for Machine Learning & AI
Implement best practices for building AI-enabled products
Develop AI-based systems for Natural Language Processing (NLP)
Optimize Machine Learning models for time efficiency, performance, cost, scalability, and accuracy
Develop tools and processes for automatically train, updating and evaluate LLM (Large Language Models)

Qualifications:

Strong foundation in Machine Learning (ML), Deep Learning, LLMs and NLP
Hands-on experience in applying Natural Language Processing solutions to challenging real-world problems.
Ability to work cross-functionally & interface with data science experts across all Eightfol's customer base
Familiar with LLM (Large Language Models), transformers like BERT, GPTs, T-5, HuggingFace etc.
Exceptionally strong knowledge of CS fundamental concepts and ML languages ( like Python, C, C++, Java, JavaScript, R, and Scala, etc. )
Ability to innovate, as proven by a track record of software artifacts or academic publications in applied machine learning.
Prior experience building and deploying machine learning models in production at scale
Understanding of data and ML systems with the ability to think across stack layers - REST APIs, microservices, data ingestion and processing systems, and distributed systems.
Extensive experience with scientific libraries in Python (numba, pandas) and machine learning tools and frameworks (scikit-learn, tensorflow, torch, etc.).
Experience implementing production machine learning systems, working with large-scale datasets, and a solid understanding of machine learning theory.
Familiar with a cloud-based environment such as AWS, Azure or GCP

Preferred Qualifications:

Metrics-focused and passionate about delivering high-quality models.
Experience with analyzing large data sets, using Hadoop, Spark 
Familiar with Spark, MLLib, Databricks MLFlow, Apache Airflow and similar related technologies.
Familiarity with MLOps tools and pipelines (MLflow, Metaflow).
PhD or Masters in Computer Science or Data Science is preferred

Micron

Salt Lake City, UT

It's about memory

The world is moving to a new economic model, where data is driving value creation in ways we had not imagined just a few years ago. Data is today’s new business currency, and memory and storage are emerging as strategic differentiators that will redefine how we extract value from data to learn, explore, communicate and experience.

And we know memory

As the leader in innovative memory solutions, Micron is helping the world make sense of data by delivering technology that is transforming how the world uses information. Through our global brands — Micron, Crucial and Ballistix — we offer the industry’s broadest portfolio. We are the only company manufacturing today’s major memory and storage technologies: DRAM, NAND, NOR and 3D XPoint™ memory. Our solutions are purpose built to leverage the value of data to unlock financial insights, accelerate scientific breakthroughs and enhance communication around the world.

It’s how we do business

From our roots in Boise, Idaho, Micron has grown into an influential global presence committed to being the best memory company in the world. This means conducting business with integrity, accountability, and professionalism and supporting our global community.

Our commitment to sustainable practices

Micron’s approach to sustainability is a reflection of how we have done business for 40 years. Our dedication to employee health and safety, environmental quality and community support worldwide is detailed in our 2019 Sustainability Report.

Our history of giving back started in 1999 when we established the Micron Foundation and identified science, technology, engineering and math (STEM) education as a priority for Idaho students. Today, the Micron Foundation’s initiatives have become leading resources for educators and students in Idaho communities and beyond, and we donate millions of dollars to organizations and events identified by our local teams around the world.

Micron Foundation

Endowed with gifts from Micron Technology, Inc., the Micron Technology Foundation’s mission is to develop effective programs that promote math, science, and engineering education and to participate in activities that address the priorities and concerns of the communities where Micron employees live and work.

Manufacturing locations

Boise, Idaho, USA
Hiroshima, Japan
Lehi, Utah, USA (IMFT, joint venture)
Manassas, Virginia, USA
Muar, Malaysia
Singapore
Taichung, Taiwan
Taoyuan, Taiwan
Xian, China
All Micron Locations

 

Similar Jobs