Principal Applied Scientist, Neuron ARG

1 month ago


Austin, Texas, 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 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 architexctures.

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.

A day in the life
Diverse Experiences
AWS values diverse experiences.

Even if you do not meet all of the qualifications and skills listed in the job description, we encourage candidates to apply.

If your career is just starting, hasn't followed a traditional path, or includes alternative experiences, don't let it stop you from applying.

Why AWS?
Amazon Web Services (AWS) is the world's most comprehensive and broadly adopted cloud platform.

We pioneered cloud computing and never stopped innovating - that's why customers from the most successful startups to Global 500 companies trust our robust suite of products and services to power their businesses.

Inclusive Team Culture
Here at AWS, it's in our nature to learn and be curious. Our employee-led affinity groups foster a culture of inclusion that empower us to be proud of our differences.

Ongoing events and learning experiences, including our Conversations on Race and Ethnicity (CORE) and AmazeCon (gender diversity) conferences, inspire us to never stop embracing our uniqueness.

Mentorship & Career Growth
We're continuously raising our performance bar as we strive to become Earth's Best Employer.

That's why you'll find endless knowledge-sharing, mentorship and other career-advancing resources here to help you develop into a better-rounded professional.

Work/Life Balance
We value work-life harmony.

Achieving success at work should never come at the expense of sacrifices at home, which is why we strive for flexibility as part of our working culture.

When we feel supported in the workplace and at home, there's nothing we can't achieve in the cloud.

Hybrid Work
We value innovation and recognize this sometimes requires uninterrupted time to focus on a build. We also value in-person collaboration and time spent face-to-face.

Our team affords employees options to work in the office every day or in a flexible, hybrid work model near one of our U.S.

Amazon offices.

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.

  • Austin, Texas, United States BAE Systems USA Full time

    Job OverviewBAE Systems USA's FAST Labs is at the forefront of scientific and technological innovation, tackling some of the most challenging issues in the defense, aerospace, power and propulsion, and security sectors. About FAST LabsThis research and development (R&D) division, part of BAE Systems' Electronic Systems sector, collaborates with various...

  • Principal Scientist

    13 hours ago


    Austin, Texas, United States Exponent Full time

    About ExponentWe are a leading engineering and scientific consulting firm that provides expert solutions to complex challenges. Our multidisciplinary teams of science, engineering, and regulatory experts work together to empower our clients with innovative solutions that create a safer, healthier, and more sustainable world.Our OpportunityWe are seeking a...


  • Austin, Texas, United States Amazon Full time

    Job OverviewPosition: Senior Applied Scientist, Last Mile OperationsAmazon's Last Mile Team is seeking a dedicated professional with exceptional optimization and analytical capabilities to join our Last Mile Science division. This role focuses on enhancing and refining the intricate planning of our global delivery network.About the TeamThe Last Mile Science...


  • Austin, Texas, United States EnergyX Full time

    Company OverviewEnergy Exploration Technologies is dedicated to becoming a global leader in the shift towards sustainable energy solutions. The organization, established in 2018, is innovating the methods by which the world harnesses and stores clean energy through advanced lithium-ion technologies and energy storage systems.Position SummaryThe Principal...


  • Austin, Texas, United States BAE Systems USA Full time

    Job DescriptionBAE Systems USA's FAST Labs is a research and development business that innovates science, technology, and engineering breakthroughs for the defense, aerospace, power, and propulsion, and security industries. Our team collaborates with business areas across the company's global enterprise to develop and implement advanced technology...


  • Austin, Texas, United States Chevrolet Full time

    About the RoleThe Data Science Manager in Marketing Applied Sciences will lead a talented team of data scientists who will drive solution design, development, and deployment of statistical and AI/ML models to solve complex business problems and improve decision making.Key ResponsibilitiesLead and manage a team of data scientists, providing guidance,...


  • Austin, Texas, United States MassGenics Full time

    MassGenics is currently seeking a Medical Lab Scientist II for the 3rd ShiftPosition Type: Full Time, ContractDuration: 06+ Months contract to hireShift Schedule: 4 x 10 9PM- 7:30 AMJob Responsibilities:Color Vision and TB Screening are mandatoryVaccination requirements include Flu and COVIDExperience as a Clinical Lab Scientist, MLT, or MT is essentialBlood...


  • Austin, Texas, United States MassGenics Full time

    MassGenics is seeking a Medical Lab Scientist II for the 3rd ShiftPosition Type: Full Time, ContractDuration: 06+ Months contract to hireLocation: Austin, TXShift Schedule: 4 x 10 9PM- 7:30 AMJob Responsibilities:Required Color Vision and TB ScreeningVaccination requirements: Flu and COVIDExperience as Clinical Lab Scientist, MLT, or MT is essentialBlood...


  • Austin, Texas, United States MassGenics Full time

    MassGenics is seeking a Medical Lab Scientist II for the 3rd ShiftPosition Type: Full Time, ContractDuration: 06+ Months contract to hireLocation: Austin, TXShift Schedule: 4 x 10 9PM- 7:30 AMJob Responsibilities:Color Vision and TB Screening are mandatoryVaccination requirements include Flu and COVIDPrior experience as a Clinical Lab Scientist, MLT, or MT...


  • Austin, Texas, United States The Texas A&M Transportation Institute (TTI) Full time

    Job Title Senior Transportation Planning Scientist/Engineer Agency Texas A&M Transportation Institute Department Transportation Planning Program Proposed Minimum Salary Commensurate Job Location Austin, Texas Job Type Faculty About Us The Texas A&M Transportation Institute (TTI) is a prominent agency within the State of Texas and a member of The Texas A&M...


  • Austin, Texas, United States Amazon Full time

    About the RoleWe are seeking an experienced Economist to join our central data and science organization for Amazon Marketing. As a key member of our team, you will develop innovative solutions to measure the return on marketing investments and drive business growth.Key ResponsibilitiesApply your expertise in causal inference and machine learning to develop...


  • Austin, Texas, United States Far Harbor, LLC Full time

    Position Overview: The Principal Social Scientist / Researcher works collaboratively with a dedicated team that designs and executes social science investigations, including program assessments. Our primary focus areas are in education and public health, and we are eager to expand our portfolio. We are looking for an individual with over 10 years of...

  • Principal R&D Engineer

    2 months ago


    Austin, Texas, United States Ansys Full time

    RESPONSIBILITIESThe Principal R&D Engineer will contribute to the development of software products and supporting systems. • Collaborate with a team of expert professionals to understand customer requirements and accomplish development objectives.Perform moderately complex development activities, including the design, implementation, maintenance, testing...


  • Austin, Texas, United States The Texas A&M Transportation Institute (TTI) Full time

    Job Title Senior Transportation Planning Scientist/Engineer Agency Texas A&M Transportation Institute Department Transportation Planning Program Proposed Minimum Salary Commensurate Job Location Austin, Texas Job Type Faculty About Us The Texas A&M Transportation Institute (TTI) serves as a vital agency within the State of Texas and is a member of The Texas...


  • Austin, Texas, United States Cadence Design Systems, Inc. Full time

    Cadence Design Systems, Inc. has openings for Principal Application Engineers, Customer Engagement (various levels/types/multiple positions) in Austin, TX. All positions require travel to various unanticipated sites throughout the U.S. Some positions may allow for telecommuting.To Apply:Send resume w/ Ref. #B202408B to . Utilizing engineering background,...

  • Senior Data Scientist

    3 weeks ago


    Austin, Texas, United States ReUp Education Full time

    What we doFounded in 2015, ReUp Education is the only organization that focuses exclusively on helping colleges and universities engage and re-enroll the more than 40 million US residents who have "stopped out" and support them until graduation, through our technology-enabled service. To date, we have re-enrolled nearly 28,500 students and recaptured over...


  • Austin, Texas, United States ClosedLoop Full time

    OverviewFor the majority of 2024, ClosedLoop has been developing a new, not-yet-launched, consumer product aimed at helping people navigate their health, built on top of generative AI.We are hiring an experienced Lead or Principal-level Product Manager to spearhead this effort.Our ideal candidate is someone who has:Experience navigating a consumer product to...


  • Austin, Texas, United States Pacific Northwest National Laboratory Full time

    Overview The Electricity Infrastructure and Buildings Division of PNNL is accelerating the transition to a decarbonized, efficient, resilient, and secure energy system through basic and applied research. We leverage a strong technical foundation in power and energy systems and in advanced data analytics to drive innovation, transform markets, and shape...


  • Austin, Texas, United States University of Texas at Austin Full time

    Job SummaryWe are seeking a highly motivated Postdoctoral Research Fellow to join our team at the University of Texas at Austin. The successful candidate will conduct research in the Department of Mechanical Engineering, focusing on mobile manipulation for handling materials in nuclear and/or hazardous environments.Key ResponsibilitiesConduct research in...


  • Austin, Texas, United States Natera Full time

    POSITION SUMMARY:The licensed Clinical Laboratory Scientist role focuses on the examination of clinical samples, ensuring that laboratory equipment is properly maintained, and supervising daily lab activities. The ideal candidate will possess a robust understanding of clinical laboratory techniques, outstanding technical and communication abilities, and the...