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

Amazon Web Services (AWS) · 15 hours ago

Applied Scientist, AWS Neuron Science Team

positionSanta Clara, CA
timeFull-time
remoteOnsite
senioritySenior Level
money$136K/yr - $223K/yr

Amazon Web Services (AWS) is seeking talented scientists for the AWS Neuron Science Team to enhance their software stack and accelerate the adoption of Trainium and Inferentia accelerators. The role involves collaborating with engineering teams and engaging with research communities to advance machine learning systems.

ConsultingDevOpsInformation TechnologySoftwareWeb Development
check
H1B Sponsor Likelynote

Responsibilities

Enhance our software stack, accelerating customer adoption of Trainium and Inferentia accelerators
Work directly with external and internal customers to identify key adoption barriers and optimization opportunities
Collaborate closely with our engineering teams to implement innovative solutions
Engage with academic and research communities to advance state-of-the-art ML systems
Develop and apply ML/RL approaches for kernel/code generation and optimization
Create advanced compiler techniques for ML workloads
Build tools for accuracy and reliability validation
Design high-performance kernels optimized for our ML accelerator architectures

Qualification

Machine LearningDeep LearningHigh-performance computingProgramming in PythonUnix/LinuxAlgorithmsData structuresNumerical optimizationData miningJavaC++MxNetTensorFlowSoft skills

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

Equity
Sign-on payments
Medical
Financial
Other benefits

Company

Amazon Web Services (AWS)

twittertwittertwitter
Glassdoor3.8
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.
dateFounded in 2002
location
Seattle, Washington, USA
people-size10001+ employees
web-link
http://aws.amazon.com

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 (18969)
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

Zawya.com
HUMAIN receives first shipment of latest NVIDIA AI chips
2025-12-31
Mexico News Daily
Mexico’s year in review: The 10 biggest business and economics stories of 2025
2025-12-31
Nextgov.com
2026 is set to be the year of agentic AI, industry predicts
2025-12-30
Company data provided by crunchbase