Current jobs related to Software Engineer II, ML Tools, Annapurna Labs - Cupertino, California - Amazon Development Center U.S., Inc.


  • 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 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 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 Amazon Full time

    Job DescriptionAnnapurna 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...


  • 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 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 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 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 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 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 Annapurna Labs (U.S.) Inc. Full time

    Job Title: Software Development Manager for ML AccelerationJoin Amazon Web Services (AWS) as a Software Development Manager for ML Acceleration and lead a team of engineers to design and deploy ML applications on various frameworks such as PyTorch, JAX, and TensorFlow.About the RoleWe are seeking an experienced Software Development Manager to lead our ML...


  • 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 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 (U.S.) Inc. Full time

    About the RoleWe are seeking a highly skilled Software Development Manager to lead our 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 strong team of engineers to design...


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

    Job SummaryWe are seeking a highly skilled Senior Quality Assurance Engineer to join our AWS Inferentia team. As a key member of our team, you will be responsible for designing and developing comprehensive test strategies and QA methodologies to ensure the quality and reliability of our ML hardware/software platform.Key Responsibilities:Design and develop...


  • 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.Key ResponsibilitiesDesign and optimize networking solutions for Machine Learning (ML) and High-Performance Computing...

Software Engineer II, ML Tools, Annapurna Labs

2 months ago


Cupertino, California, United States Amazon Development Center U.S., Inc. Full time
About the Role

We are seeking a highly skilled Software Engineer to lead the development of machine learning tools at Amazon's Annapurna Labs. 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 Responsibilities
  • Lead the design and implementation of machine learning tools to run, optimize, and analyze workloads
  • Collaborate with cross-functional teams to drive business decisions and deliver high-impact solutions to our large customer base
  • Participate in design discussions, code review, and communicate with internal and external stakeholders
  • Work in a startup-like development environment, where you'll always be working on the most important stuff
About the Team

Our team is dedicated to supporting new members and fostering a culture of knowledge-sharing and mentorship. We have a broad mix of experience levels and tenures, and we're building an environment that celebrates career growth and development.

What We Offer
  • A dynamic and collaborative work environment
  • Opportunities for career growth and development
  • A comprehensive benefits package
  • A competitive salary and bonus structure
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
Preferred Qualifications
  • 3+ years of full software development life cycle experience
  • Bachelor's degree in computer science or equivalent
  • Experience taking a leading role in building complex software systems
  • Strong proficiency in Go, Python, and Javascript/Typescript