Principal Applied Scientist, Neuron ARG (Cupertino) Job at Amazon Web Services (AWS), Cupertino, CA

ZzhrSWZ5OVpyWmxXS0dONG9xeGJCSFV0RGc9PQ==
  • Amazon Web Services (AWS)
  • Cupertino, CA

Job Description

Join to apply for the Principal Applied Scientist, Neuron ARG role at Amazon Web Services (AWS)

1 week ago Be among the first 25 applicants

Join to apply for the Principal Applied Scientist, Neuron ARG role at Amazon Web Services (AWS)

Get AI-powered advice on this job and more exclusive features.

Description

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.

Description

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 an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status.

Los Angeles County applicants: Job duties for this position include: work safely and cooperatively with other employees, supervisors, and staff; adhere to standards of excellence despite stressful conditions; communicate effectively and respectfully with employees, supervisors, and staff to ensure exceptional customer service; and follow all federal, state, and local laws and Company policies. Criminal history may have a direct, adverse, and negative relationship with some of the material job duties of this position. These include the duties and responsibilities listed above, as well as the abilities to adhere to company policies, exercise sound judgment, effectively manage stress and work safely and respectfully with others, exhibit trustworthiness and professionalism, and safeguard business operations and the Company’s reputation. Pursuant to the Los Angeles County Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records.

Our inclusive culture empowers Amazonians to deliver the best results for our customers. If you have a disability and need a workplace accommodation or adjustment during the application and hiring process, including support for the interview or onboarding process, please visit for more information. If the country/region you’re applying in isn’t listed, please contact your Recruiting Partner.

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.

Company - Amazon Development Center U.S., Inc.

Job ID: A2728641

Seniority level

  • Seniority level

    Director

Employment type

  • Employment type

    Full-time

Job function

  • Job function

    Research, Science, and Engineering
  • Industries

    IT Services and IT Consulting

Referrals increase your chances of interviewing at Amazon Web Services (AWS) by 2x

Get notified about new Principal jobs in Cupertino, CA .

Mountain View, CA $204,500.00-$276,500.00 2 weeks ago

San Mateo, CA $180,000.00-$300,000.00 1 week ago

San Francisco Bay Area $250,000.00-$350,000.00 2 weeks ago

Sr. Manager, Head of Developer Relations

San Jose, CA $200,000.00-$250,000.00 6 days ago

Menlo Park, CA $472,000.00-$472,000.00 2 weeks ago

San Jose, CA $200,000.00-$225,000.00 4 days ago

San Jose, CA $255,000.00-$300,000.00 2 weeks ago

Head of ML/AI Exploration, Architecture, & Engineering

Mountain View, CA $248,000.00-$345,000.00 1 day ago

Mountain View, CA $216,000.00-$312,550.00 1 week ago

San Francisco Bay Area $225,000.00-$275,000.00 2 weeks ago

Vice President, Engineering - Data Science and Data Engineering Products

Santa Clara, CA $480,000.00-$690,000.00 1 day ago

Head of Growth Engineering and Operations

Head of Strategic Talent Solutions, MSP & Growth

San Mateo, CA $175,000.00-$225,000.00 2 months ago

Pleasanton, CA $161,100.00-$286,200.00 1 week ago

We’re unlocking community knowledge in a new way. Experts add insights directly into each article, started with the help of AI.

#J-18808-Ljbffr

Job Tags

Full time, Work at office, Local area, Flexible hours,

Similar Jobs

Royal Caribbean Group

Senior Manager, Security Facilities Job at Royal Caribbean Group

 ...Cruises the most innovative fleet and private destinations, and the best people. Together...  .... Royal Caribbean Groups Safety, Security, Quality and Environmental Team has an...  ...role, the Sr. Manager will liaison with contractors and facilities maintenance to provide... 

ImpactSearch Partners

Treatment Coordinator Job at ImpactSearch Partners

 ...Performs a variety of reception, secretarial, insurance, and data entry duties for our patients...  ..., schedules large treatment, and coordinates patient flow. Promotes a safe, pleasant...  ...daily work. JOB SPECIFICATIONS: Dental or business experience High school graduate... 

Macpower Digital Assets Edge

Senior Systems Safety Engineer/Nuclear Engineer Job at Macpower Digital Assets Edge

 ...Job Summary: Our expected ideal candidate would be from Mid to Senior Systems Safety Engineer/Nuclear Engineer with experience performing Nuclear Surety/Safety for weapons systems. Preferred experience is 8+ years. BASIC QUALIFICATIONS: Requires a Bachelor... 

A to Z Drain Service

Bath Installer Job at A to Z Drain Service

 ...What you will do: Bath and shower installer will estimate and perform accurate and sleek construction projects relating to the Bathroom remodeling. Install and finish drywall, painting, floor repairs and installation, framing, plumbing and basic installation on vanities... 

AbsoluteCare

Treatment & Triage Nurse - PRN Job at AbsoluteCare

 ...Remote?No Why Work at AbsoluteCare? At...  ...care team of doctors, nurses, social workers, and medical...  ...The Treatment and Triage Nurse offers AbsoluteCare...  ...Room (ER) by performing telephonic and in-person triage,...  ...in the community, their home, or our comprehensive care...