Software Development Engineer II, Annapurna Labs

1 week ago


Cupertino, California, United States Amazon Full time
Job Description

Annapurna Labs is a leading innovator in custom Machine Learning accelerators, and we're seeking a talented Software Development Engineer to join our team. As a key member of our Neuron Compiler Engineering team, you will play a critical role in developing the infrastructure of a compiler to enable efficient execution of large-scale ML workloads in the cloud.

The ideal candidate will have a strong background in software development, with experience in designing and implementing scalable systems. You will work closely with our cross-functional team to design, implement, and test innovative software solutions that transform service performance, durability, cost, and security.

Key Responsibilities:

  • Design and implement efficient and reliable build, test, and release mechanisms for the Neuron compiler
  • Participate in design discussions, code review, and communication with internal and external stakeholders
  • Collaborate with a cross-functional team to drive business decisions with technical input
  • Work in a startup-like development environment, where you're always working on the most important stuff

Requirements:

  • 3+ years of non-internship professional software development experience
  • 2+ years of non-internship design or architecture experience
  • Experience programming with at least one software programming language

What We Offer:

  • A competitive salary range of $129,300/year to $223,600/year, depending on location
  • Equity, sign-on payments, and other forms of compensation
  • A full range of medical, financial, and other benefits

Amazon is an Equal Opportunity Employer:

We are committed to a diverse and inclusive workplace. Amazon is an equal opportunity employer and does not discriminate on the basis of race, national origin, gender, gender identity, sexual orientation, protected veteran status, disability, age, or other legally protected status.



  • Cupertino, California, United States Annapurna Labs Full time

    Job SummaryAt Annapurna Labs, we're seeking a talented Software Development Engineer to join our team. As a key member of our engineering team, you will be responsible for innovating and delivering creative software designs to develop new services, solve operational problems, and drive improvements in developer velocity.Key ResponsibilitiesInnovate and...


  • Cupertino, California, United States Annapurna Labs (U.S.) Inc. Full time

    About the RoleWe are seeking a highly skilled Senior Runtime Software Development Engineer to join our team at Annapurna Labs (U.S.) Inc. in the AWS AI organization. As a key member of our team, you will be responsible for designing and developing innovative software solutions for our machine learning accelerators, including the AWS Inferentia and Trainium...


  • Cupertino, California, United States Annapurna Labs (U.S.) Inc. Full time

    About the RoleWe are seeking a highly skilled Senior Runtime Software Development Engineer to join our team at Annapurna Labs (U.S.) Inc. in the AWS AI organization. As a key member of our team, you will be responsible for designing, developing, and delivering high-performance software solutions for our machine learning accelerators.Key...


  • Cupertino, California, United States Amazon Full time

    About the RoleAnnapurna Labs, a part of Amazon, is seeking a talented Software Development Engineer to join our Neuron Compiler team. As a key member of our team, you will be responsible for designing and implementing innovative software solutions to transform service performance, durability, cost, and security.Key ResponsibilitiesSolve challenging technical...


  • Cupertino, California, United States Amazon Full time

    About the RoleAWS is seeking a highly skilled Software Engineer III to lead the development of machine learning tools to run, optimize, and analyze machine learning workloads. As a member of the Annapurna Labs team, you will design and implement new tools, pipelines, and automation to ensure compatibility with existing and next-generation AI accelerators.Key...


  • Cupertino, California, United States Annapurna Labs Full time

    About the RoleAnnapurna Labs, now fully integrated into AWS, is a leader in infrastructure innovation. Our Machine Learning Applications (ML Apps) team is seeking a skilled Software Engineer to join our AWS Neuron team. As a key member of this team, you will be responsible for developing, enabling, and performance-tuning a wide range of machine learning...


  • Cupertino, California, United States Annapurna Labs (U.S.) Inc. Full time

    About the JobWe are seeking a highly skilled Software Development Manager to lead our Machine Learning (ML) Applications Framework team. As a key member of our organization, you will be responsible for designing and deploying ML applications and use cases on various frameworks such as PyTorch, JAX, and TensorFlow.Key ResponsibilitiesLead a team of engineers...


  • Cupertino, California, United States Amazon Full time

    About the RoleWe are seeking a highly skilled Senior Software Development Engineer to join our Nitro Card team at Annapurna Labs. As a key member of our team, you will be responsible for designing and developing software for our next-generation EC2 instances.The ideal candidate will have a strong background in software development, with experience in C, C++,...


  • Cupertino, California, United States Amazon Full time

    About the RoleWe are seeking a highly skilled Senior Modeling Engineer to lead our SoC Modeling Team in Annapurna Labs, a part of AWS. As a key member of our team, you will be responsible for architecting and delivering new functional and performance models, infrastructure, and tooling for our customers.Key ResponsibilitiesChart the technical direction for...


  • Cupertino, California, United States Amazon Full time

    About the RoleWe are seeking an experienced Embedded Software Development Engineer to join our Annapurna Nitro team. As a key member of our team, you will be responsible for designing, implementing, and testing software for our Nitro Card platform.Key ResponsibilitiesResearch and design firmware to support various hardware units in the Nitro cardDebug...


  • Cupertino, California, United States Amazon Full time

    About the RoleWe are seeking a highly skilled Machine Learning Engineer II to join our Annapurna ML pathfinding team. As a key member of this team, you will be responsible for helping our most strategic customers port their models to the AWS Trainium & Inferentia platforms.Key ResponsibilitiesDeliver high-quality code and customizations to make models...


  • Cupertino, California, United States Amazon Development Center U.S., Inc. Full time

    About the RoleWe are seeking a highly skilled Software Engineer II to lead the development of machine learning tools for our ML Accelerators. As a key member of our team, you will design and implement new toolsets, collaborate with developers, system architects, and hardware engineers to ensure compatibility with existing and next-generation AI...


  • Cupertino, California, United States Annapurna Labs Full time

    About the RoleWe are seeking a highly skilled Software Development Manager to lead our Machine Learning Acceleration team. As a key member of our organization, you will be responsible for designing and deploying ML applications and use cases on various frameworks such as PyTorch, JAX, and TensorFlow.As the Software Development Manager for the ML Applications...


  • Cupertino, California, United States Amazon Development Center U.S., Inc. Full time

    About the RoleWe are seeking a highly skilled Software Engineer to lead the development of machine learning tools for our AI accelerators. As a key member of our team, you will design and implement new toolsets, collaborate with developers, system architects, and hardware engineers to ensure compatibility with existing and next-generation AI accelerators.Key...


  • Cupertino, California, United States Amazon Full time

    About the RoleWe are seeking an experienced software engineer with expertise in low-latency networking and collective operations to optimize customer experience by designing systems that enable scaling network-intensive workloads over thousands of CPUs, GPUs, and TPUs.This role is on the forefront of AI/ML, where we spend a good deal of time optimizing the...


  • Cupertino, California, United States Annapurna Labs Full time

    About the RoleWe are seeking a highly experienced Senior Software Development Manager to lead our Neuron Inference Customer Enablement Team. As a key member of our organization, you will be responsible for optimizing customer or open-source models for inference performance on various frameworks such as PyTorch, JAX, and TensorFlow.Key ResponsibilitiesLead a...


  • Cupertino, California, United States Amazon Full time

    About the RoleWe are seeking an experienced software engineer with expertise in low-latency networking to optimize customer experience by designing systems that enable scaling network-intensive workloads over thousands of CPUs, GPUs, and TPUs. This role is on the forefront of AI/ML, where we spend a good deal of the day optimizing the networking for the...


  • Cupertino, California, United States Amazon Full time

    About the RoleWe are seeking a highly skilled ASIC Physical Design Methodology Engineer to join our team at Annapurna Labs, a part of Amazon Web Services (AWS). As a member of our Cloud-Scale Machine Learning Acceleration team, you will be responsible for designing and optimizing hardware in our data centers, including AWS Inferentia and Trainium...


  • Cupertino, California, United States Amazon Full time

    About the RoleWe are seeking a highly skilled Senior Software Development Engineer to join our Machine Learning (ML) Infrastructure team. As a key member of this team, you will be responsible for designing and developing the tools and infrastructure that support the success of our ML and High Performance Computing (HPC) technologies.As a Senior Software...


  • Cupertino, California, United States Annapurna Labs Full time

    Job SummaryWe are seeking a highly skilled Quality Assurance Engineer to join our AWS Inferentia cloud-scale machine learning accelerator team. As a key member of this team, you will be responsible for testing and validating the performance of our cutting-edge hardware and software platform.Key ResponsibilitiesDevelop and execute test plans and strategies to...