Applied Scientist, Neuron ARG, Annapurna ML

Amazon • Seattle, WA • 57min ago

Description

The Automated Reasoning Group in the Amazon Neuron 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 models across diverse domains, including Large Language and Vision, originating from leading frameworks such as PyTorch and JAX. Your role will involve working closely with our custom-built Machine Learning accelerator, Trainium, which represents the forefront of innovation for advanced ML capabilities, and is the underpinning of Generative AI.

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 architect and implement business-critical tooling, publish 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, theorem provers, and program synthesis engines will be a benefit in this role.

A background in machine learning and AI accelerators is preferred but not required.

Basic Qualifications

PhD, or Master's degree and 4+ years of CS, CE, ML or related field experience

Experience in patents or publications at top-tier peer-reviewed conferences or journals

Experience programming in Java, C++, Python or related language

Experience in any of the following areas: algorithms and data structures, parsing, numerical optimization, data mining, parallel and distributed computing, high-performance computing

Preferred Qualifications

PhD in computer science, computer engineering, or related field

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 https://amazon.jobs/content/en/how-we-hire/accommodations for more information. If the country/region you're applying in isn't listed, please contact your Recruiting Partner.

The base salary range for this position is listed below. Your Amazon package will include sign-on payments and restricted stock units (RSUs). Final compensation will be determined based on factors including experience, qualifications, and location. Amazon also offers comprehensive benefits including health insurance (medical, dental, vision, prescription, Basic Life & AD&D insurance and option for Supplemental life plans, EAP, Mental Health Support, Medical Advice Line, Flexible Spending Accounts, Adoption and Surrogacy Reimbursement coverage), 401(k) matching, paid time off, and parental leave. Learn more about our benefits at https://amazon.jobs/en/benefits .

USA, CA, Cupertino - 171,600.00 - 222,200.00 USD annually

USA, WA, Seattle - 142,800.00 - 193,200.00 USD annually

Amazon

Go to Apply

Share this job

Similar Jobs

Executive Business Administrator

Microsoft Corporation • Redmond, WA • 43min ago

Overview The Engineering & Architecture Group (EAG) within Industry Solutions Delivery (ISD) and Microsoft Frontier Company is a global organization of architects, engineers, data scientists, security experts, design...

43min ago

Go to Apply

Applied Scientist, Worldwide Grocery Stores - Data and Science

Amazon • Seattle, WA • 45min ago

Retail & Hospitality

Description Amazon's Worldwide Grocery Stores (WWGS), Data & Science team is seeking an Applied Scientist to join our Sales & Operations Planning (S&OP) and Supply Chain Science team. In this role, you will h...

45min ago

Go to Apply

Applied Scientist III, Sponsored Products and Brands - Advertiser Growth - Cross-border Seller Experience (CBSX)

Amazon • Seattle, WA • 52min ago

Description The Sponsored Products and Brands team at Amazon Ads is re-imagining the advertising landscape through generative AI technologies, revolutionizing how millions of customers discover products and engage with b...

52min ago

Go to Apply

SDE - II, Search Product Grouping And Discovery Team

Amazon • Seattle, WA • 58min ago

IT & Engineering

Description Every day, hundreds of millions of customers come to Amazon Search looking for something, sometimes they know exactly what they want, sometimes they're just browsing. The Search Experience team at Amazon Shop...

58min ago

Go to Apply

Applied Scientist, Alexa Enterprise

Amazon • Seattle, WA • 58min ago

Description Are you excited about building at the intersection of enterprise AI and customer experience? Are you passionate about applying science to real customer problems? Do you want your work shaped by real-world con...

58min ago

Go to Apply

Sr. Applied Scientist, Alexa Enterprise

Amazon • Seattle, WA • 58min ago

58min ago

Go to Apply

Sr. Applied Scientist, Alexa Enterprise

Amazon • Seattle, WA • 58min ago

58min ago

Go to Apply

Career Fair Applications

Employee Owned Holdings, Inc. • Seattle, WA • 7h ago

Employee Owned Holdings, Inc. (EOH) is a holding company of three operating companies in the hydraulic and fluid power industry: Hydraquip - distributor of hydraulic and electrical systems Gulf Controls Company (GCC) - d...

7h ago

Go to Apply

Purchasing Buyer

Rainier Industries • Seattle, WA • 7h ago

Retail & Hospitality

OAI + Rainier by The Look Company, is a manufacturing organization with a long, established history. We are seeking a detail-oriented Purchasing Buyer to manage end-to-end procurement activities and support production ac...

7h ago

Go to Apply

Software Engineer 2

Microsoft Corporation • Redmond, WA • 7h ago

IT & Engineering

Overview Microsoft is a company where passionate innovators come to collaborate, envision what can be, and take their careers further. This is a world of more possibilities, more innovation, more openness, and the sky is...

7h ago

Go to Apply

Subscribe to get new jobs sent directly to you
(no charge, cancel anytime)