Software Engineer AI/ML, AWS Neuron Distributed Training Team

2 weeks ago


Cupertino, California, United States Annapurna Labs Full time
About the Role

Annapurna 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 model families, including large language models and vision transformers.

About the Team

Our team is dedicated to creating innovative solutions for our customers. We believe in fostering an inclusive and diverse work environment, where everyone feels valued and empowered to contribute. We offer flexible work arrangements, mentorship opportunities, and a commitment to career growth and development.

Responsibilities

* Develop integrations with the Neuron SDK and frameworks such as TensorFlow, PyTorch, and MXNet
* Plan and implement new features, working with customers to create innovative solutions
* Contribute to open-source projects and collaborate with other teams to drive innovation

Requirements

* B.S. in Computer Science or related technical field
* Experience with one or more of the following programming languages: C, C++, Java, or Perl
* Experience with deep learning frameworks: TensorFlow, PyTorch, and MXNet

Preferred Qualifications

* M.S. or Ph.D. in Computer Science, Deep Learning, Artificial Intelligence, Applied Math, or related field
* Experience with distributed training for Deep Learning and High Performance Computing

What We Offer

* Competitive salary and benefits package
* Opportunities for career growth and development
* Flexible work arrangements and work-life balance

Amazon is an Equal Opportunity Employer

Amazon is committed to a diverse and inclusive workplace. We are an equal opportunity employer and do 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 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 Amazon Full time

    About the RoleWe are seeking a highly skilled Senior Software Engineer to join our AWS Neuron Applications team. As a key member of our team, you will be responsible for developing, enabling, and performance tuning of a wide variety of machine learning model families, including large language models, stable diffusion, and vision transformers.Our team works...


  • Cupertino, California, United States Amazon Full time

    About the RoleThe Sr. Software Development Manager of ML Frameworks & Ecosystems at AWS will lead the team that develops and extends Neuron support for leading ML frameworks including PyTorch and JAX.This role requires a strong technical understanding and a motivation to achieve results, with experience partnering with product or program management teams and...


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

    About the RoleWe are seeking a highly skilled Software Development Engineer to join our AWS Neuron Inference team. As a key member of this team, you will be responsible for developing, enabling, and optimizing a wide range of machine learning models, including large language models, vision transformers, and more.Key ResponsibilitiesDesign and develop...


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

    Job DescriptionAs a Senior Software Engineer on the Apple Intelligence team, you will play a key role in developing groundbreaking conversational assistant technologies for large-scale systems and new client devices. You will work closely with cross-functional teams to design, implement, and evaluate AI/ML systems, ensuring they meet the highest standards of...


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

    About the RoleThis is a software engineer position in the Machine Learning Applications (ML Apps) team for AWS Neuron. The team works on development, enablement, and performance tuning of machine learning models, including large language models and vision transformers.The ideal candidate will have experience optimizing inference performance for latency and...


  • Cupertino, California, United States Amazon Full time

    About the Role:The AWS Neuron team is seeking a highly skilled Senior Machine Learning Compiler Engineer to join our team. As a key member of our team, you will be responsible for designing and developing a compiler to handle the world's largest ML workloads. You will work closely with our ML services teams to ensure that our compiler meets the needs of our...


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

    About the RoleAs a Software Development Manager for Machine Learning Acceleration, you will be responsible for leading a team of engineers to design and deploy machine learning applications and use cases on various frameworks such as PyTorch, JAX, and TensorFlow. You will be responsible for the full development life cycle of our integrations and extensions...


  • Cupertino, California, United States Amazon Full time

    Job DescriptionThe AWS Neuron Compiler team is seeking a skilled compiler engineer to develop a state-of-the-art deep learning compiler stack.This stack optimizes application models across diverse domains, including Large Language and Vision, originating from leading frameworks like PyTorch, TensorFlow, and JAX.The successful candidate will work closely with...


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

    Job SummaryWe are seeking a highly skilled Senior Software Engineer to join our dynamic team in Apple's ML Platform & Technology group. This role will focus on designing, building, and maintaining large-scale distributed systems to support the end-to-end machine learning (ML) lifecycle.Key ResponsibilitiesDesign and develop large-scale distributed systems...


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

    About the RoleAs a Senior Software Engineer on the Apple Intelligence team, you will play a key part in building groundbreaking conversational assistant technologies for large-scale systems and new client devices. Our team is dedicated to creating innovative solutions that delight and inspire millions of people worldwide.Key Responsibilities Own...


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