Principal Applied Scientist, Neuron ARG

23 hours ago


Seattle WA United States Amazon Full time

Job ID: 2728641 | Amazon Development Center U.S., Inc.

The AWS Neuron Compiler team is actively seeking applied scientists to join our efforts in developing a state-of-the-art deep learning compiler stack. This stack is designed to optimize application models across diverse domains, including Large Language and Vision, originating from leading frameworks such as PyTorch, TensorFlow, and JAX. Your role will involve working closely with our custom-built Machine Learning accelerators, including Inferentia/Trainium, which represent the forefront of AWS innovation for advanced ML capabilities, powering solutions like Generative AI.

The Automated Reasoning Group in the AWS Neuron Compiler team works on the intersection of Artificial Intelligence and program analysis to raise the code quality bar in compiler stack. In this role as an Applied Scientist, you'll be instrumental in designing, developing, and deploying analyzers for ML compiler stages and compiler IRs. You will contribute to the design and semantics of DSLs that our customers love when crafting neural architectures. You will architect and implement business-critical tooling, publish cutting-edge research, and mentor a brilliant team of experienced scientists and engineers. You will need to be technically capable, credible, and curious in your own right as a trusted AWS Neuron engineer, innovating on behalf of our customers. Your responsibilities will involve tackling crucial challenges alongside a talented engineering team, contributing to leading-edge design and research in compiler technology and deep-learning systems software. Strong experience in programming languages, compilers, program analyzers, and program synthesis engines will be a benefit in this role.

A background in machine learning and AI accelerators is preferred but not required. We expect a strong publication record in compilers, IR semantics, static analysis or verification.

BASIC QUALIFICATIONS

* PhD in computer science or a related field
* A strong publication record
* Track record in hands-on compiler or tool development

PREFERRED QUALIFICATIONS

* Publications in programming languages, compilers, static analysis or verification
* A track record of mentoring junior researchers
* A track record of working with a wider community of compiler users and developers

Amazon is committed to a diverse and inclusive workplace. Amazon is an equal opportunity employer and does not discriminate on the basis of race, national origin, gender, gender identity, sexual orientation, protected veteran status, disability, age, or other legally protected status. For individuals with disabilities who would like to request an accommodation, please visit

Our compensation reflects the cost of labor across several US geographic markets. The base pay for this position ranges from $179,000/year in our lowest geographic market up to $309,400/year in our highest geographic market. Pay is based on a number of factors including market location and may vary depending on job-related knowledge, skills, and experience. Amazon is a total compensation company. Dependent on the position offered, equity, sign-on payments, and other forms of compensation may be provided as part of a total compensation package, in addition to a full range of medical, financial, and/or other benefits. For more information, please visit This position will remain posted until filled. Applicants should apply via our internal or external career site.

#J-18808-Ljbffr

  • Seattle, WA, United States Amazon Full time

    Job ID: 2728641 | Amazon Development Center U.S., Inc. The AWS Neuron Compiler team is actively seeking applied scientists to join our efforts in developing a state-of-the-art deep learning compiler stack. This stack is designed to optimize application models across diverse domains, including Large Language and Vision, originating from leading frameworks...


  • Seattle, WA, United States Amazon Full time

    The AWS Neuron Compiler team is actively seeking applied scientists to join our efforts in developing a state-of-the-art deep learning compiler stack. This stack is designed to optimize application models across diverse domains, including Large Language and Vision, originating from leading frameworks such as PyTorch, TensorFlow, and JAX. Your role will...


  • Seattle, WA, United States Amazon Full time

    The AWS Neuron Compiler team is actively seeking applied scientists to join our efforts in developing a state-of-the-art deep learning compiler stack. This stack is designed to optimize application models across diverse domains, including Large Language and Vision, originating from leading frameworks such as PyTorch, TensorFlow, and JAX. Your role will...


  • Seattle, WA, United States Amazon Full time

    Job ID: 2714956 | Amazon Development Center U.S., Inc. The Automated Reasoning Group in the AWS Neuron Compiler team is looking for an Applied Scientist to work on the intersection of Artificial Intelligence and program analysis to raise the code quality bar in our state-of-the-art deep learning compiler stack. This stack is designed to optimize application...


  • Seattle, United States Amazon Development Center U.S., Inc. Full time

    Minimum Qualifications PhD in computer science or a related field A strong publication record Track record in hands-on compiler or tool development The AWS Neuron Compiler team is actively seeking applied scientists to join our efforts in developing a state-of-the-art deep learning compiler stack. This stack is designed to optimize application models...


  • Seattle, WA, United States Amazon Full time

    Principal Applied Scientist, AGI Foundations Job ID: 2762609 | Amazon.com Services LLC As a Principal Scientist within the Artificial General Intelligence (AGI) organization, you are a trusted part of the technical leadership. You bring business and industry context to science and technology decisions.You set the standard for scientific excellence and make...


  • Seattle, WA, United States Amazon Full time

    Principal Applied Scientist, Amazon Private Brands Job ID: 2835385 | Amazon.com Services LLC The Amazon Private Brands’ Discovery team develops innovative machine learning solutions to enhance customer awareness of Amazon's private brands and help customers discover products they love. This interdisciplinary team of scientists and engineers incubates and...


  • Seattle, WA, United States Amazon Full time

    Principal Applied Scientist, AGI Data Services Job ID: 2815583 | Amazon.com Services LLC The Artificial General Intelligence Data Service (AGI-DS) team is seeking a dedicated, skilled, and innovative Principal Applied Scientist with a robust background in deep learning, to build industry-leading technology with Large Language Models (LLMs) and multimodal...


  • Seattle, WA, United States Amazon Full time

    Job ID: 2846094 | Amazon.com Services LLC Amazon Prime’s Science organization leads the Research & Development towards innovation for Amazon Prime. Amazon Prime is the backbone of Amazon’s consumer business and aspires to be the world’s most engaging, satisfying, and loved membership program, driving growth and profitability. The program serves over...


  • Seattle, WA, United States Amazon Full time

    Sr. Principal Applied Scientist, Sponsored Products, Advertiser Growth Job ID: 2785228 | Amazon Advertising LLC The Sponsored Products (SP) team is looking for a Sr. Principal Scientist to help our millions of advertisers create, manage and optimize their campaigns with improved recommendations to drive efficient advertising sales. Amazon Advertising is one...


  • Bellevue, WA, United States Amazon Full time

    Principal Applied Scientist, Optimal Inventory Health Job ID: 2623476 | Amazon.com Services LLC Amazon's Optimal Inventory Health (OIH) org in Supply Chain Optimization (SCOT) group is looking for a Principal Applied Scientist to optimize one of the most complex eCommerce systems in the world. The Optimal Inventory Health (OIH) drives long-term cash flow...


  • Seattle, WA, United States Amazon Full time

    Principal Applied Scientist - CV/ML, Amazon Robotics Job ID: 2843016 | Amazon.com Services LLC Do you want to create worldwide impact in robotics while solving challenges at the edge of robotics research? Our team in Amazon Robotics builds high-performance, real-time robotic systems that can perceive, learn, and act intelligently alongside humans—at Amazon...


  • Sunnyvale, CA, United States Amazon Full time

    Job ID: 2829026 | Amazon.com Services LLC As a Principal Scientist in Ring and Blink, you will have deep subject matter expertise in the areas of Artificial Intelligence. You will work with multiple teams of scientists and engineers to translate business and functional requirements into concrete deliverables. More specifically, you will provide thought...


  • Seattle, WA, United States Amazon Full time

    Are you a scientist interested in pushing the state of the art in Generative AI, LLMs, LMMs? Are you interested in working on ground-breaking research projects that will lead to great products and scientific publications? Do you wish you had access to large datasets? Answer yes to any of these questions and you’ll fit right in here at Amazon.We are looking...


  • Seattle, United States Amazon.com Services LLC Full time

    Job Title: Principal Applied ScientistMinimum Qualifications:MS degree (or equivalent) in Electrical Engineering, Computer Science, Mathematics, or related technical field5+ years of industrial/academic product experience, such as formal verification, program analysis, constraint-solving, and theorem proving5+ years of technical management experience5+ years...


  • Seattle, WA, United States Amazon Full time

    Principal Applied Scientist - Live Playback and Operations, Live Playback and Operations Live Playback and Operations organization owns the live and 24/7 linear channel playback experience from content ingestion and signal acquisition, encoding (H.264/AVC, H.265/HEVC, AV1 etc) and streaming technologies (DASH, HLS, Smooth streaming, Low Latency), feature...


  • Seattle, WA, United States Amazon Full time

    Job ID: 2615083 | Amazon.com Services LLC Are you a scientist interested in pushing the state of the art in Generative AI, LLMs, LMMs? Are you interested in working on ground-breaking research projects that will lead to great products and scientific publications? Do you wish you had access to large datasets? Answer yes to any of these questions and you’ll...


  • Seattle, WA, United States Amazon Full time

    Join an innovative team of scientists and engineers who use machine learning and statistical techniques to create state-of-the-art solutions for providing better value to Amazon's customers. Key job responsibilities Amazon Science gives you insight into the company’s approach to customer-obsessed scientific innovation. Amazon fundamentally believes that...


  • Seattle, United States Amazon Full time

    Principal Applied Scientist, Amazon Private BrandsJob ID: 2835385 | Amazon.com Services LLCThe Amazon Private Brands’ Discovery team develops innovative machine learning solutions to enhance customer awareness of Amazon's private brands and help customers discover products they love. This interdisciplinary team of scientists and engineers incubates and...


  • Seattle, United States Amazon Full time

    Principal Applied Scientist, AGI Data ServicesJob ID: 2815583 | Amazon.com Services LLCThe Artificial General Intelligence Data Service (AGI-DS) team is seeking a dedicated, skilled, and innovative Principal Applied Scientist with a robust background in deep learning, to build industry-leading technology with Large Language Models (LLMs) and multimodal...