Software Development Manager for ML Acceleration

1 week ago


Cupertino, California, United States Annapurna Labs (U.S.) Inc. Full time
About the Role

We 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 Responsibilities
  • Lead a strong team of engineers to design and deploy ML applications and use cases on various frameworks.
  • Develop reliability, scalability features, and performance updates in the Neuron ML Frameworks.
  • Contribute to other popular open frameworks to enable them to work seamlessly with Trainium and Inferentia devices.
  • Partner with product and program management teams to define product directions and deliver them to customers.
Requirements
  • 3+ years of engineering team management experience.
  • 7+ years of working directly within engineering teams experience.
  • 3+ years of designing or architecting new and existing systems experience.
  • Experience partnering with product or program management teams.
Preferred Qualifications
  • 2 years of experience leading teams that develop machine learning models using PyTorch and/or TensorFlow.
  • Experience working with large distributed fleets of GPU or other accelerated systems.
About Us

Amazon Web Services (AWS) is the world's most comprehensive and broadly adopted cloud platform. We pioneered cloud computing and continue to innovate, providing a robust suite of products and services to power businesses.

Why Join Us

We value diverse experiences and encourage candidates from all backgrounds to apply. We offer a flexible work environment, opportunities for growth and development, and a commitment to diversity, equity, and inclusion.



  • Cupertino, California, United States Apple Full time

    About the RoleWe are seeking a highly skilled Senior Engineering Program Manager to lead the development and deployment of GPU-accelerated machine learning solutions across Apple's product ecosystem.Key ResponsibilitiesLead multi-functional teams to develop and deploy GPU-accelerated ML solutionsDrive the product development lifecycle from concept through...


  • Cupertino, California, United States Apple Full time

    About the RoleWe are seeking a highly skilled Senior Engineering Program Manager to lead the development and deployment of GPU-accelerated machine learning solutions across Apple's product ecosystem.Key ResponsibilitiesLead multi-functional teams to develop and deploy GPU-accelerated ML solutionsDrive the product development lifecycle from concept through...


  • Cupertino, California, United States Amazon Full time

    About the RoleWe are seeking a highly experienced Senior Software Development Manager to lead our team in developing and extending Neuron support for leading ML frameworks, including PyTorch and JAX. As a key member of our AWS Neuron team, you will be responsible for delivering framework plugins and libraries that enable a great user experience for...


  • Cupertino, California, United States Apple Full time

    Job Title: Senior Engineering Program Manager, GPU AccelerationJob Summary:We are seeking a highly skilled Senior Engineering Program Manager to lead the development of GPU-accelerated machine learning solutions across Apple's product ecosystem. The successful candidate will have experience in supporting machine learning accelerated by GPU technology and be...


  • Cupertino, California, United States Apple Full time

    About the RoleWe are seeking a highly skilled Senior Engineering Program Manager to lead the development of GPU-accelerated machine learning solutions across Apple's product ecosystem.Key ResponsibilitiesLead multi-functional teams to develop and deploy GPU-accelerated ML solutionsDrive the product development lifecycle from concept through launch, ensuring...


  • Cupertino, California, United States Apple Full time

    About the RoleWe are seeking a highly skilled Senior Engineering Program Manager to lead the development of GPU-accelerated machine learning solutions across Apple's product ecosystem.Key ResponsibilitiesLead multi-functional teams to develop and deploy GPU-accelerated ML solutionsDrive the product development lifecycle from concept through launch, ensuring...


  • Cupertino, California, United States Apple Full time

    About the RoleWe are seeking a highly skilled Senior Engineering Program Manager to lead the development and deployment of GPU-accelerated machine learning solutions across Apple's product ecosystem.Key ResponsibilitiesLead multi-functional teams to develop and deploy GPU-accelerated ML solutionsDrive the product development lifecycle from concept through...


  • Cupertino, California, United States Apple Full time

    About the RoleWe are seeking a highly skilled Senior Engineering Program Manager to lead the development and deployment of GPU-accelerated machine learning solutions across Apple's product ecosystem.Key ResponsibilitiesLead multi-functional teams to develop and deploy GPU-accelerated ML solutionsDrive the product development lifecycle from concept through...


  • Cupertino, California, United States Apple Full time

    Role OverviewWe are seeking a highly skilled engineer to join our Platform Architecture Software Performance Optimization Team. As a key member of this team, you will work on analyzing existing and new workloads to identify performance bottlenecks in hardware and/or software.Key ResponsibilitiesAnalyze performance challenges on Apple silicon and emerging new...


  • Cupertino, California, United States Apple Full time

    About the RoleWe are seeking a highly skilled Senior Engineering Program Manager to lead the development and deployment of GPU-accelerated machine learning solutions across Apple's product ecosystem.Key ResponsibilitiesLead multi-functional teams to develop and deploy GPU-accelerated ML solutionsDrive the product development lifecycle from concept through...


  • Cupertino, California, United States Apple Full time

    Role OverviewWe are seeking a highly skilled engineer to join our Platform Architecture Software Performance Optimization Team. As a key member of our team, you will work on analyzing existing and new workloads to identify performance bottlenecks in hardware and/or software.Key ResponsibilitiesAnalyze performance challenges on Apple silicon and emerging new...


  • Cupertino, California, United States Apple Inc. Full time

    Job SummaryWe are seeking an experienced Software Engineering Manager to lead our Applied ML Engineering team in the Software Localization department at Apple Inc. The successful candidate will be responsible for managing a diverse team of Applied ML engineers who will work cross-functionally and globally to deliver the next generation of ML tooling and...


  • Cupertino, California, United States Apple Full time

    About the RoleWe are seeking a highly skilled Software Development Engineer in Test to join our ML Systems Evaluation Engineering team at Apple. As a key member of this team, you will play a critical role in evaluating and qualifying Siri and other AI and ML-based products to ensure they meet the highest standards of quality and user experience.Key...


  • Cupertino, California, United States Apple Full time

    About the RoleWe are seeking a highly skilled Software Development Engineer in Test to join our ML Systems Evaluation Engineering team at Apple. As a key member of this team, you will play a critical role in evaluating and ensuring the quality of our AI and ML-based products, including Siri and other Apple ecosystem features.Key ResponsibilitiesDesign and...


  • Cupertino, California, United States Apple Inc. Full time

    Technical Leadership for Machine Learning LocalizationAt Apple Inc., we're seeking a seasoned technical leader to manage a diverse team of Applied Machine Learning engineers. This role is part of our Software Localization team, which powers the tooling, features, and processes used to create an exceptional localized experience for our international...


  • Cupertino, California, United States Apple Full time

    Job SummaryWe are seeking a highly skilled Software Development Engineer in Test to join our ML Systems Evaluation Engineering team at Apple. As a key member of our team, you will be responsible for evaluating and qualifying AIML products, including Siri and other Apple user experiences, to ensure they meet the highest standards of quality and innovation.Key...


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

    Role SummaryWe are seeking a highly skilled Software Development Engineer in Test to join our ML Systems Evaluation Engineering team at Apple. As a key member of this team, you will play a critical role in evaluating and qualifying AI/ML products, including Siri and other AIML technologies.Key ResponsibilitiesDevelop automation and test methodologies to...


  • Cupertino, California, United States Amazon Full time

    About the RoleWe are seeking a highly experienced Senior Software Development Manager to lead our team responsible for the development of Machine Learning Collective Operations on the AWS Trainium platforms. The ideal candidate will have a strong background in HPC interconnects, preferably in the ML domain, and a good understanding of collective...


  • Cupertino, California, United States Amazon Full time

    About the RoleWe are seeking a highly skilled Senior Machine Learning Engineer to join our Annapurna ML pathfinding team. As a key member of this team, you will play a critical role in helping our most strategic customers accelerate their adoption of Annapurna ML products, including AWS Trainium and AWS Inferentia.Key ResponsibilitiesCollaborate with...