ZipRecruiter

Log In

1

Junior Machine Learning Compiler Engineer Jobs in California

Nvidia

Senior AI Compiler Engineer

Santa Clara, CA

$122K - $168K/yr

We are seeking a Machine Learning Compiler Engineer with deep expertise in compiler technologies to join our team. The ideal candidate will bring broad experience across the machine learning ...

next page

Showing results 1-20

All JobsMachine Learning Compiler Engineer JobsJunior Machine Learning Compiler Engineer JobsJunior Machine Learning Compiler Engineer Jobs in California

Junior Machine Learning Compiler Engineer information

What are typical projects and responsibilities for a Junior Machine Learning Compiler Engineer in a collaborative team setting?

As a Junior Machine Learning Compiler Engineer, you can expect to work on projects that focus on optimizing machine learning models for performance and deployment across various hardware platforms. Typical responsibilities include assisting in developing and debugging compiler passes, implementing optimizations, and contributing to code reviews. You'll frequently collaborate with senior engineers, data scientists, and hardware specialists to ensure that models are efficiently translated and executed. This role offers valuable learning opportunities through hands-on coding, exposure to state-of-the-art ML frameworks, and regular team meetings for knowledge sharing and mentorship.

What does a Junior Machine Learning Compiler Engineer do?

A Junior Machine Learning Compiler Engineer helps design, develop, and optimize compilers for machine learning models. Their work involves translating high-level machine learning code into efficient low-level code that can run on various hardware platforms, such as CPUs, GPUs, or specialized AI chips. They often collaborate with software engineers and data scientists to ensure that machine learning workloads run efficiently and correctly. This role typically involves programming, debugging, and performance tuning, often using languages like C++, Python, and specialized frameworks.

What are the key skills and qualifications needed to thrive as a Junior Machine Learning Compiler Engineer, and why are they important?

To thrive as a Junior Machine Learning Compiler Engineer, you need a solid background in computer science fundamentals, programming (especially C++ and Python), and foundational knowledge of machine learning and compiler theory. Familiarity with frameworks and tools such as LLVM, TensorFlow, MLIR, and version control systems is typically required, along with a relevant bachelor’s or master’s degree. Strong problem-solving abilities, attention to detail, and effective teamwork and communication skills set standout candidates apart. These skills and qualities are crucial for efficiently optimizing machine learning models for various hardware targets and collaborating on innovative compiler solutions.

What is the difference between Junior Machine Learning Compiler Engineer vs Data Scientist?

AspectJunior Machine Learning Compiler EngineerData Scientist
Required CredentialsBachelor's in Computer Science, Software Engineering, or related field; knowledge of compiler design and ML frameworksBachelor's or higher in Data Science, Statistics, Computer Science, or related field; strong analytical skills
Work EnvironmentSoftware development teams, focusing on compiler optimization for ML modelsData analysis teams, focusing on data interpretation and model development
Employer & Industry UsageTech companies, AI startups, hardware firmsTech firms, finance, healthcare, research institutions

The Junior Machine Learning Compiler Engineer primarily focuses on developing and optimizing compilers for machine learning models, requiring programming and compiler knowledge. In contrast, a Data Scientist analyzes data, builds models, and provides insights. Both roles are essential in AI and tech industries but differ in technical focus and daily tasks.

What are the most commonly searched types of Machine Learning Compiler Engineer jobs in California? The most popular types of Machine Learning Compiler Engineer jobs in California are:
What are popular job titles related to Junior Machine Learning Compiler Engineer jobs in California? For Junior Machine Learning Compiler Engineer jobs in California, the most frequently searched job titles are:
What job categories do people searching Junior Machine Learning Compiler Engineer jobs in California look for? The top searched job categories for Junior Machine Learning Compiler Engineer jobs in California are:
What cities in California are hiring for Junior Machine Learning Compiler Engineer jobs? Cities in California with the most Junior Machine Learning Compiler Engineer job openings:
Junior Machine Learning Compiler Engineer jobs near you
Sr. Machine Learning - Compiler Engineer III, AWS Neuron, Annapurna Labs

Sr. Machine Learning - Compiler Engineer III, AWS Neuron, Annapurna Labs

Amazon

Cupertino, CA • On-site

$128K - $177K/yr

Full-time

Posted 23 days ago

Amazon rating

7.4

Company rating: 7.4 out of 10

Based on 6,843 frontline employees who took The Breakroom Quiz

6th of 39 rated national retailers

Job description

The Product: AWS Machine Learning accelerators are at the forefront of AWS innovation and one of several AWS tools used for building Generative AI on AWS. The Inferentia chip delivers best-in-class ML inference performance at the lowest cost in cloud. Trainium will deliver the best-in-class ML training performance with the most teraflops (TFLOPS) of compute power for ML in the cloud.

This is all enabled by cutting edge software stack, the AWS Neuron Software Development Kit (SDK), which includes an ML compiler, runtime and natively integrates into popular ML frameworks, such as PyTorch, TensorFlow and MxNet. AWS Neuron and Inferentia are used at scale with customers like Snap, Autodesk, Amazon Alexa, Amazon Rekognition and more customers in various other segments.
The Team: As a whole, the Amazon Annapurna Labs team is responsible for silicon development at AWS. The team covers multiple disciplines including silicon engineering, hardware design and verification, software and operations


The AWS Neuron team works to optimize the performance of complex neural net models on our custom-built AWS hardware. More specifically, the AWS Neuron team is developing a deep learning compiler stack that takes neural network descriptions created in frameworks such as TensorFlow, PyTorch, and MXNET, and converts them into code suitable for execution. As you might expect, the team is comprised of some of the brightest minds in the engineering, research, and product communities, focused on the ambitious goal of creating a toolchain that will provide a quantum leap in performance.
You: As a Sr

Machine Learning Compiler Engineer III on the AWS Neuron team, you will be a thought leader supporting the ground-up development and scaling of a compiler to handle the world's largest ML workloads. Architecting and implementing business-critical features, publish cutting-edge research, and mentoring a brilliant team of experienced engineers excites and challenges you. You will leverage your technical communications skill as a hands-on partner to AWS ML services teams and you will be involved in pre-silicon design, bringing new products/features to market, and many other exciting projects.

A background in Machine Learning and AI accelerators is preferred, but not required.
In order to be considered for this role, candidates must be currently located or willing to relocate to Cupertino (perferred), Seattle, Austin, or Toronto.
About the team
Inclusive Team Culture
Here at AWS, we embrace our differences. We are committed to furthering our culture of inclusion. We have ten employee-led affinity groups, reaching 40,000 employees in over 190 chapters globally

We have innovative benefit offerings, and host annual and ongoing learning experiences, including our Conversations on Race and Ethnicity (CORE) and AmazeCon (gender diversity) conferences. Amazon's culture of inclusion is reinforced within our 16 Leadership Principles, which remind team members to seek diverse perspectives, learn and be curious, and earn trust.
Work/Life Balance
Our team puts a high value on work-life balance. It isn't about how many hours you spend at home or at work; it's about the flow you establish that brings energy to both parts of your life

We believe striking the right balance between your personal and professional life is critical to life-long happiness and fulfillment. We offer flexibility in working hours and encourage you to find your own balance between your work and personal lives.
Mentorship & Career Growth
Our team is dedicated to supporting new members. We have a broad mix of experience levels and tenures, and we're building an environment that celebrates knowledge sharing and mentorship

We care about your career growth and strive to assign projects based on what will help each team member develop into a better-rounded professional and enable them to take on more complex tasks in the future.

What Amazon employees say

Pay

Benefits

Hours and flexibility

Workplace

Get the full story on Breakroom

Amazon logo

About Amazon

Sourced by ZipRecruiter

Amazon.com, Inc., commonly known as Amazon, is an American multinational technology company. It was founded by Jeff Bezos in 1994 and initially started as an online marketplace for books. Since then, Amazon has expanded its operations and become one of the largest e-commerce companies in the world. Amazon's primary business is its online retail platform, where customers can purchase a vast array of products, including electronics, clothing, books, home goods, and much more. The company offers a convenient and user-friendly shopping experience, with features such as fast shipping, customer reviews, and personalized recommendations. In addition to its e-commerce platform, Amazon has diversified its business into various other areas. One of its notable ventures is Amazon Web Services (AWS), a comprehensive cloud computing platform that provides services such as storage, compute power, and database management to individuals and businesses. AWS has become a leader in the cloud computing industry, powering many websites and applications worldwide. Amazon has also developed its own consumer electronics, including the popular Amazon Kindle e-reader, Fire tablets, Fire TV streaming devices, and the Alexa-powered Echo smart speakers. The Alexa voice assistant, integrated into these devices, allows users to interact with their devices using voice commands, perform tasks, and access information. Furthermore, Amazon has expanded into media and entertainment. It operates Prime Video, a streaming service that offers a wide range of movies, TV shows, and original content. Amazon Music provides a platform for streaming and purchasing digital music, while Audible offers audiobooks and other audio content. The company's commitment to customer satisfaction and convenience is demonstrated by its membership program, Amazon Prime. Prime members receive various benefits, including free two-day shipping, access to streaming services, exclusive deals, and more.

Industry

It services, book publishers, retail, real estate and computer and electronic product manufacturing

Company size

10,000+ Employees

Headquarters location

Seattle, WA, US

View All Amazon Jobs

Apply