Sr. Machine Learning Compiler Engineer, AWS Neuron, Annapurna Labs

2 weeks ago


Seattle, United States Amazon Web Services (AWS) Full time

Join to apply for the Sr. Machine Learning Compiler Engineer, AWS Neuron, Annapurna Labs role at Amazon Web Services (AWS)1 day ago Be among the first 25 applicantsDescriptionThe Product: AWS Machine Learning accelerators are at the forefront of AWS innovation and are used for building Generative AI on AWS. The Inferentia chip delivers top ML inference performance at the lowest cloud cost. Trainium provides the best ML training performance with high TFLOPS. This is enabled by the AWS Neuron SDK, which includes an ML compiler, runtime, and integration with frameworks like PyTorch, TensorFlow, and MxNet. AWS Neuron and Inferentia are used by customers such as Snap, Autodesk, Amazon Alexa, and Amazon Rekognition.The TeamThe Amazon Annapurna Labs team develops silicon for AWS, covering silicon engineering, hardware design, verification, software, and operations. The AWS Neuron team focuses on optimizing neural net model performance on AWS hardware, developing a deep learning compiler stack that translates framework descriptions into executable code. The team aims to create a toolchain that significantly boosts performance.Your RoleAs a Sr. Machine Learning Compiler Engineer III, you will lead the development and scaling of a compiler for large ML workloads, architect and implement critical features, publish research, and mentor engineers. You will collaborate with AWS ML services teams and be involved in pre-silicon design, product launches, and other projects. Experience in ML and AI accelerators is preferred but not required.Candidates must be willing to relocate to Seattle.About The TeamWe value inclusion, with employee-led affinity groups, benefits, and diversity conferences. Our culture emphasizes learning, trust, and diverse perspectives.We prioritize work-life balance, offering flexible hours. We support career growth through mentorship and project assignments that develop professional skills.Basic Qualifications5+ years of professional software development experienceProficiency in at least one programming languageExperience in system design, reliability, and scalingFull software development lifecycle experienceExperience mentoring or leading engineering teamsPreferred QualificationsBachelor's degree in computer science or equivalentAmazon is an equal opportunity employer. For accommodation support, visit this link. Compensation varies by location and experience, with additional benefits. The position remains open until filled. Apply via our careers site. #J-18808-Ljbffr



  • Seattle, United States Amazon Web Services (AWS) Full time

    Join to apply for the Sr. Machine Learning - Compiler Engineer III, AWS Neuron, Annapurna Labs role at Amazon Web Services (AWS).Get AI-powered advice on this job and more exclusive features.DescriptionThe Product: AWS Machine Learning accelerators are at the forefront of AWS innovation and are used for building Generative AI on AWS. The Inferentia chip...


  • Seattle, Washington, United States Amazon Web Services (AWS) Full time $151,300 - $261,500 per year

    DescriptionThe Product: AWS Machine Learning accelerators are at the forefront of AWS innovation and one of several AWS tools used for building Generative AI on AWS. The Inferentia chip delivers best-in-class ML inference performance at the lowest cost in cloud. Trainium will deliver the best-in-class ML training performance with the most teraflops (TFLOPS)...


  • Seattle, United States Annapurna Labs Full time

    Annapurna Labs builds custom Machine Learning accelerators that are at the forefront of AWS innovation and one of several AWS tools used for building Generative AI on AWS. The Neuron Compiler Engineering team is searching for a Senior Software Development Engineer to support the development infrastructure of a compiler to enable the world's largest ML...


  • Seattle, United States Annapurna Labs Full time

    Annapurna Labs builds custom Machine Learning accelerators that are at the forefront of AWS innovation and one of several AWS tools used for building Generative AI on AWS. The Neuron Compiler Engineering team is searching for a Senior Software Development Engineer to support the development infrastructure of a compiler to enable the world's largest ML...


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

    Annapurna Labs builds custom Machine Learning accelerators that are at the forefront of AWS innovation and one of several AWS tools used for building Generative AI on AWS. The Neuron Compiler Engineering team is searching for a Senior Software Development Engineer to support the development infrastructure of a compiler to enable the world's largest ML...


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

    Annapurna Labs builds custom Machine Learning accelerators that are at the forefront of AWS innovation and one of several AWS tools used for building Generative AI on AWS. The Neuron Compiler Engineering team is searching for a Senior Software Development Engineer to support the development infrastructure of a compiler to enable the world's largest ML...


  • Seattle, United States Amazon Full time

    The Product: AWS Machine Learning accelerators are at the forefront of AWS innovation and one of several AWS tools used for building Generative AI on AWS. The Inferentia chip delivers best-in-class ML inference performance at the lowest cost in cloud. Trainium will deliver the best-in-class ML training performance with the most teraflops (TFLOPS) of compute...


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

    AWS Machine Learning accelerators are at the forefront of AWS innovation. The Trainium chip delivers industry-leading ML inference and training performance at the lowest cost in the cloud. This is enabled by edge software stack, the AWS Neuron Software Development Kit, which includes a Machine Learning compiler. The SDK natively integrates into popular ML...


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

    AWS Machine Learning accelerators are at the forefront of AWS innovation. The Trainium chip delivers industry-leading ML inference and training performance at the lowest cost in the cloud. This is enabled by edge software stack, the AWS Neuron Software Development Kit, which includes a Machine Learning compiler. The SDK natively integrates into popular ML...


  • Seattle, Washington, United States Amazon Web Services (AWS) Full time $151,300 - $261,500 per year

    DescriptionAnnapurna Labs builds custom Machine Learning accelerators that are at the forefront of AWS innovation and one of several AWS tools used for building Generative AI on AWS. The Neuron Compiler Engineering team is searching for a Senior Software Development Engineer to support the development infrastructure of a compiler to enable the world's...