Sr Software Dev Engineer, Deep Learning Compilers

Job Description

Join us to accelerate deep learning networks on Az1/Az2 Neural Edge processors and beyond, delivering solutions to offload Speech, Computer Vision, and Generative AI workloads on various devices.

Requirements

• 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 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
• 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

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
• Collaborate with hardware, software, applied science, and product teams to onboard user experiences to be powered by Deep Learning accelerator
• Mentor and provide guidance to junior engineers

Benefits

• Medical, dental, and vision coverage (multiple plan options)
• Health savings and flexible spending accounts
• 24/7 Employee Assistance Program for mental health support
• 401(k) with company match and various investment choices
• Company-paid life/AD&D and disability insurance
• Restricted stock units (RSUs) for employee ownership
• Child care and elder care referrals, adoption assistance
• Paid maternity and parental leave (including Ramp Back)
• Paid time off (PTO) and company holidays
• Employee discount on Amazon.com products
• Career Choice program (95% tuition and textbook coverage)