Amazon Sr Software Dev Engineer Deep Learning Compilers 

United States, California, Sunnyvale 

604216663

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
- 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, develop new tooling where necessary
- Participate in design reviews, API development, and documentation- Mentor and provide guidance to junior engineers

- 5+ years of non-internship professional software development experience
- 5+ years of programming with at least one software programming language experience
- 5+ years of leading design or architecture (design patterns, reliability and scaling) of new and existing systems experience
- Experience as a mentor, tech lead or leading an engineering team
- Experience with neural network inference offload to GPU, DSP or custom accelerators

- 5+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience
- Bachelor's degree in computer science or equivalent
- Experience building compiler for application specific accelerators or custom instruction set
- Knowledge of compute offload using compiler frameworks such as TVM, PyTorch, XLA, MLIR