Amazon.com Services LLC

Technology

SoftwareDevEngineer,MachineLearningCompilers

$165–224k Sunnyvale, California, United States FULL TIME
Market Sentiment
HIGH DEMAND

Neural analysis suggests this role is
optimal for Mid+ candidates.

The Brief

“Software Dev Engineer, Machine Learning Compilers at Amazon.com Services LLC. Skills: Machine Learning Compilers, Deep Learning, Edge AI, Custom Silicon. Design software stack. Develop compiler passes”

Industry & Context.

Technology
Problems you'll solve

Root cause analysis; Debugging; Performance analysis; Troubleshooting

What They're Looking For.

Must Have

2+ years systems design experience, 3+ years software development experience, 3+ years programming experience, Experience developing LLMs

Nice to Have

3+ years full software development life cycle, Experience with compilers, Experience with program analysis, Experience with optimization algorithms, Experience with computer architecture, Experience with machine learning systems, Experience with embedded software, Experience with low-level programming, Experience with C, Experience with C++, Experience with Python, Experience with Linux, Experience with performance analysis, Experience with debugging, Experience with system performance tuning, Experience with distributed systems, Experience with cloud platforms, Experience with AI hardware, Experience with custom silicon, Experience with edge AI, Experience with neural accelerators, Experience with deep learning networks, Experience with model quantization, Experience with model compression, Experience with heterogeneous device backends, Experience with software stack development, Experience with compiler passes, Experience with graph ingestion, Experience with graph optimization, Experience with graph partitioning, Experience with backend code generation, Experience with system level performance optimization, Experience with tooling development, Experience with design reviews, Experience with API development, Experience with documentation, Experience with hardware teams, Experience with software teams, Experience with applied science teams, Experience with product teams, Experience with user experiences, Experience with AI agents, Experience with research teams, Experience with interface design, Experience with silicon development, Experience with consumer devices, Experience with LLMs on GPUs, Experience with LLMs on Neuron, Experience with LLMs on TPUs

What You'll Do.

Design software stack

Develop compiler passes

Develop backend code generation

Profile system performance

Analyze system performance

Optimize system performance

Participate in design reviews

Collaborate with hardware teams

Collaborate with software teams

Collaborate with applied science teams

Collaborate with product teams

Onboard user experiences

Mentor junior engineers

Provide guidance to junior engineers

Develop compiler infrastructure

Lower deep learning workloads

Innovate on model quantization

Innovate on model compression

Bring AI models to silicon

Own system components

Meet with research team

Sketch cleaner interfaces

How You'll Work.

Team & Collaboration

Partner science teams; Peer science teams; Hardware teams; Software teams; Applied science teams; Product teams; Research team; Hardware engineers; Firmware engineers

Process & Methodology

Design patterns

Full Job Description

Amazon Devices is an inventive research and development company that designs and engineers high-profile consumer products like the Kindle family, Fire Tablets, Fire TV, Health & Wellness devices, Amazon Echo, and Astro. We are building the next generation of edge AI capabilities through our advanced compression platform, compiler and custom neural accelerator silicon. Come join us to accelerate deep learning networks on edge processors and beyond. We are looking for a talented and passionate software engineer to be part of an exciting technology creation team at Amazon. You will have an enormous opportunity to make a large impact on the design, architecture, and implementation of deep learning technologies embedded into consumer products used every day, by people you know. The position provides an unique opportunity to contribute and make an impact from hardware design stage followed by pre and post silicon development as well as productizing it on consumer devices. In this role you will be work along side partner science teams to develop the compiler infrastructure and lower deep learning workloads to heterogeneous device backends. You will also partner up with peer science teams to innovate on model quantization and compression techniques for efficient execution on hardware. Key job responsibilities Design and develop software stack for deep learning accelerator Develop Compiler passes for graph ingestions, optimizations and partitioning. Develop backend code generation capabilities across heterogeneous platforms Profile, analyze and optimize system level performance, develop new tooling where necessary Participate in design reviews, API development, and documentation Successfully collaborate with hardware, software, applied science and product teams to onboard more and more user experiences to be powered by Deep Learning accelerator. Mentor and provide guidance to junior engineers A day in the life You join a small team building the compiler that brings large AI

Free ATS check

Applying for this Software Dev Engineer, Machine Learning Compilers role?

Most applicants get filtered before a human reads their resume. See if yours makes the cut.

ANONYMOUS · UNFILTERED

What do employees actually say about Amazon.com Services LLC?

Real rants from real employees. Read before you apply.

Read Company Rants →