Machine Learning Compiler Engineer

2 weeks ago


Cupertino, California, United States Amazon Full time
About the Role

We are seeking a highly skilled Machine Learning Compiler Engineer to join our team at Amazon. As a member of our Neuron Compiler team, you will play a key role in developing and scaling a compiler to handle the world's largest ML workloads.

Your primary responsibility will be to architect and implement business-critical features, publish cutting-edge research, and contribute to a brilliant team of experienced engineers. You will leverage your technical communications skills as a hands-on partner to AWS ML services teams.

This is an exciting opportunity to join a talented team of innovators who are pushing the boundaries of what is possible with AI and Machine Learning. If you have a passion for innovation, data, search, analytics, and distributed systems, we want to hear from you.



  • Cupertino, California, United States Amazon Full time

    Required Skills and QualificationsCurrently enrolled in a Bachelor's degree program or higher in Computer Science, Computer Engineering, Electrical Engineering, or related field.Programming experience in internship or coursework with programming languages such as Python, C, or C++.Strong interests and academic qualifications/research focus in two of the...


  • Cupertino, California, United States Amazon Full time

    Amazon is seeking a skilled Machine Learning Compiler Developer to join the AWS Neuron team.">About the RoleYou will be responsible for architecting and implementing business-critical features, publishing cutting-edge research, and contributing to a brilliant team of experienced engineers. As a Machine Learning Compiler Developer, you will leverage your...


  • Cupertino, California, United States Amazon Full time

    About the RoleWe are seeking a highly experienced Senior Machine Learning Compiler Architect to join our AWS Neuron team. As a member of this team, you will be responsible for architecting and implementing business-critical features for our deep learning compiler stack.Key ResponsibilitiesArchitect and implement key components of our deep learning compiler...


  • Cupertino, California, United States Amazon Full time

    We are seeking a highly skilled Machine Learning Compiler Engineer to join our Amazon team in the AWS Neuron division. As a key member of our team, you will be responsible for architecting and implementing business-critical features, publishing cutting-edge research, and collaborating with experienced engineers to develop a compiler that handles the world's...


  • Cupertino, California, United States Amazon Full time

    **Job Description:**A Software Development Engineer is needed in the Machine Learning Applications team for AWS Neuron. This role is responsible for development, enablement, and performance tuning of various machine learning models.The ideal candidate will have experience with distributed inference libraries such as Deepspeed and optimizing inference...


  • Cupertino, California, United States Amazon Full time

    About the Role: We are seeking a highly skilled Machine Learning Engineer to join our team in developing and optimizing machine learning models for AWS Neuron. This role will involve working closely with compiler engineers and runtime engineers to create, build, and tune distributed inference solutions.Key Responsibilities:Design and develop high-impact...


  • Cupertino, California, United States Amazon Full time

    About the JobWe are seeking a skilled Cloud Scale Machine Learning Engineer to join our team. As a key member of our Machine Learning Applications team, you will be responsible for developing, enabling, and tuning distributed inference solutions using AWS Neuron.Salary: $173,450 - $178,650 per yearKey ResponsibilitiesDesign and implement high-performance...


  • Cupertino, California, United States Amazon Full time

    Job OverviewA high-level software development engineer position is available in the Machine Learning Applications team for AWS Neuron.About the RoleThis role involves developing, enabling, and performance tuning of various machine learning model families, including large language models, stable diffusion, and vision transformers.The successful candidate will...


  • Cupertino, California, United States Amazon Full time

    Job SummaryThe AWS Neuron team is seeking a skilled Software Development Engineer to join our Machine Learning Applications team. As a key member of this team, you will be responsible for developing, enabling, and performance-tuning a wide variety of machine learning model families, including large language models and vision transformers.This role requires...


  • Cupertino, California, United States Amazon Full time

    About the RoleThis is an exciting opportunity to join the Annapurna Labs team at Amazon Web Services (AWS) as a Senior Software Engineer. We are seeking a highly skilled engineer with expertise in deep learning and distributed training. As a member of our Machine Learning Applications (ML Apps) team, you will be responsible for developing and maintaining...


  • Cupertino, California, United States Amazon Full time

    Are you passionate about designing and developing cutting-edge machine learning solutions? Do you thrive in a fast-paced environment where innovation is key?About the RoleWe are seeking an experienced Machine Learning Architect to join our team at Amazon. As a key member of our ML Applications team, you will be responsible for developing and implementing...


  • Cupertino, California, United States Amazon Full time

    About the JobWe are seeking a highly skilled Sr. Machine Learning Compiler Engineer III to join our team at Amazon in Cupertino, Seattle, Austin, or Toronto.Your primary responsibility will be to lead the ground-up development and scaling of a compiler to handle the world's largest ML workloads.You will be a thought leader supporting the design,...


  • Cupertino, California, United States Amazon Full time

    About the Role: Amazon's Machine Learning Engineering team is seeking a talented Team Lead to join our team. As a key member of our ML Apps team, you will be responsible for leading the development and deployment of large-scale machine learning models on AWS Neuron. This includes designing and implementing distributed training solutions using PyTorch,...


  • Cupertino, California, United States Amazon Full time

    OverviewThe AWS Neuron Compiler team is dedicated to developing cutting-edge compiler technology for deep learning applications. As a Senior Compiler Engineer, you will play a crucial role in enhancing the accuracy and reliability of our compiler stack.This position requires collaboration with our custom-built Machine Learning accelerators, including...


  • Cupertino, California, United States Amazon Full time

    Job DescriptionThis role is for a software engineer in the Machine Learning Applications (ML Apps) team for AWS Neuron. The team works side by side with compiler engineers and runtime engineers to create, build, and tune distributed inference solutions with Trn1.As a cloud-scale machine learning engineer, you will be responsible for optimizing inference...


  • Cupertino, California, United States Apple Full time

    About the CompanyApple is the best place to do on-device machine learning, and this team sits at the heart of that discipline, interfacing with research, SW engineering, HW engineering, and products.Job OverviewThis role requires understanding of ML operator primitives, common compiler optimizations (frontend/middle-end), runtimes, and system software...


  • Cupertino, California, United States Amazon Full time

    About the RoleThis is a unique opportunity to join Amazon's Machine Learning Applications (ML Apps) team as a software development engineer. You will be responsible for developing, enabling, and performance-tuning a wide range of machine learning models, including large language models like Llama2, GPT2, and GPT3.Your primary focus will be on optimizing...


  • Cupertino, California, United States Apple Full time

    OverviewAt Apple, we're committed to delivering innovative products and services that enrich people's lives. As a Machine Learning Engineer, you will play a critical role in developing and deploying machine learning models to drive business outcomes.About the RoleThis is an exciting opportunity to work with a talented team of engineers and scientists to...


  • Cupertino, California, United States Apple Full time

    Job OverviewWe are seeking a highly skilled Machine Learning Engineer to join our team at Apple. As an expert in deep learning, you will be responsible for developing and implementing innovative machine learning solutions for our products.


  • Cupertino, California, United States Amazon Full time

    **Role Overview:**We are seeking an experienced Software Development Engineer to join our Machine Learning Applications team for AWS Neuron. This role involves developing, enabling, and tuning machine learning models for cloud-scale applications.The ideal candidate will have strong software development skills in C++/Python and ML knowledge, with experience...