Sr Software Dev Engineer, Deep Learning Compilers

Global technology company specializing in e-commerce, cloud computing, digital streaming, and artificial intelligence.
$151,300 - $261,500
Machine Learning
Senior Software Engineer
In-Person
5,000+ Employees
5+ years of experience
AI · Enterprise SaaS

Description For Sr Software Dev Engineer, Deep Learning Compilers

Amazon is seeking a Senior Software Development Engineer specializing in Deep Learning Compilers to join their innovative team working on Az1/Az2 Neural Edge processors. This role focuses on accelerating deep learning networks and delivering solutions for Speech, Computer Vision, and Generative AI workloads across various Amazon devices, from Blink Smart Home cameras to Echo Show products.

The position offers a unique opportunity to impact the entire development lifecycle, from hardware design to post-silicon development and product implementation. You'll work closely with science teams to develop compiler infrastructure and optimize deep learning workloads for heterogeneous device backends. The role involves sophisticated technical challenges in compiler development, performance optimization, and system architecture.

As a senior engineer, you'll be responsible for architecting and developing the software stack for deep learning accelerators, creating compiler passes for graph operations, and implementing backend code generation across multiple platforms. The role requires strong leadership skills, as you'll be mentoring junior engineers and collaborating across multiple teams including hardware, software, applied science, and product development.

The compensation is highly competitive, ranging from $151,300 to $261,500 per year, depending on location and experience, plus additional benefits including equity and sign-on bonuses. This is an excellent opportunity for experienced engineers passionate about deep learning, compiler development, and hardware acceleration to make a significant impact on products used by millions of customers.

Amazon offers a comprehensive benefits package and maintains an inclusive culture that empowers employees to deliver the best results for customers. The role is based in Sunnyvale, CA, and requires at least 5 years of professional software development experience, with a strong background in neural network inference and system architecture.

Last updated 2 minutes ago

Responsibilities For Sr Software Dev Engineer, Deep Learning Compilers

  • Architect 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
  • Participate in design reviews, API development, and documentation
  • Collaborate with hardware, software, applied science and product teams
  • Mentor and provide guidance to junior engineers

Requirements For Sr Software Dev Engineer, Deep Learning Compilers

Python
  • 5+ years of non-internship professional software development experience
  • 5+ years of programming with at least one software programming language
  • 5+ years of leading design or architecture experience
  • Experience as a mentor, tech lead or leading an engineering team
  • Experience with neural network inference offload to GPU, DSP or custom accelerators
  • Bachelor's degree in computer science or equivalent (preferred)
  • Experience building compiler for application specific accelerators or custom instruction set (preferred)
  • Knowledge of compute offload using compiler frameworks such as TVM, PyTorch, XLA, MLIR (preferred)

Benefits For Sr Software Dev Engineer, Deep Learning Compilers

Medical Insurance
401k
Equity
  • Medical benefits
  • Financial benefits
  • Equity compensation
  • Sign-on payments

Interested in this job?

Jobs Related To Amazon Sr Software Dev Engineer, Deep Learning Compilers

Sr. Software Engineer- AI/ML, AWS Neuron Distributed Training

Senior Software Engineering role at AWS focusing on machine learning infrastructure, distributed training, and performance optimization for cloud-scale AI accelerators.

Sr. Machine Learning Engineer, Routing and Planning

Senior Machine Learning Engineer role at Amazon focusing on AI solutions for Last Mile delivery optimization and routing planning, handling 11B+ packages globally.

Sr. Software Engineer- AI/ML, AWS Neuron Distributed Training

Senior ML Engineering role at AWS focusing on distributed training systems and ML accelerator optimization, offering competitive pay and comprehensive benefits.

Sr. Physical Design Engineer - Static Timing Analysis, Annapurna Labs, Cloud Scale Machine Learning

Senior Physical Design Engineer role at AWS focusing on static timing analysis and machine learning acceleration, offering competitive compensation and comprehensive benefits.

Senior Software Dev Engineer, Product Quality Tech

Senior Software Engineer role at Amazon focusing on AI/ML systems to protect marketplace integrity and customer trust through product quality verification.