SIGN IN
Applied Scientist, AWS Neuron Science Team jobs in United States
cer-icon
Apply on Employer Site
company-logo

Amazon Web Services (AWS) · 8 hours ago

Applied Scientist, AWS Neuron Science Team

Amazon Web Services (AWS) is looking for talented scientists to enhance their software stack for Trainium and Inferentia accelerators. In this role, you'll collaborate with engineering teams and engage with academic communities to advance state-of-the-art ML systems.
Artificial Intelligence (AI)Cloud ComputingConsultingSoftwareInformation TechnologyAgentic AIDevOpsWeb Development
check
H1B Sponsor Likelynote

Responsibilities

Developing and applying ML/RL approaches for kernel/code generation and optimization
Creating advanced compiler techniques for ML workloads
Building tools for accuracy and reliability validation
Designing high-performance kernels optimized for our ML accelerator architectures
Supporting the development and management of Compute, Database, Storage, Platform, and Productivity Apps services in AWS

Qualification

PhD in Computer ScienceMachine Learning CompilerDeep Learning FrameworksProgramming in JavaProgramming in PythonUnix/Linux ExperienceAlgorithmsData StructuresNumerical OptimizationData MiningHigh-Performance Computing

Required

PhD in computer science, computer engineering, or related field
Experience in patents or publications at top-tier peer-reviewed conferences or journals
Experience in any of the following areas: algorithms and data structures, parsing, numerical optimization, data mining, parallel and distributed computing, high-performance computing
Experience programming in Java, C++, Python or related language
Experience using Unix/Linux

Preferred

Experience in investigating, designing, prototyping, and delivering new and innovative system solutions
Experience in state-of-the-art deep learning models architecture design and deep learning training and optimization and model pruning
Experience with popular deep learning frameworks such as MxNet and Tensor Flow

Benefits

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
Parental leave

Company

Amazon Web Services (AWS)

company-logo
Launched in 2006, Amazon Web Services (AWS) began exposing key infrastructure services to businesses in the form of web services -- now widely known as cloud computing.

H1B Sponsorship

Amazon Web Services (AWS) has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (22803)
2024 (21175)
2023 (19057)
2022 (24088)
2021 (12233)
2020 (14881)

Funding

Current Stage
Late Stage
Total Funding
unknown
Key Investors
BIRD Foundation
2025-01-22Grant

Leadership Team

leader-logo
Matt Garman
Chief Executive Officer
linkedin
leader-logo
Anand Desikan
CTO, CXO Advisor, and Enterprise Technologist
linkedin

Recent News

Company data provided by crunchbase