Software Engineer II, ML Tools, Annapurna Labs

6 hours 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 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 Responsibilities
  • Lead the design and implementation of machine learning tools for AI accelerators
  • Collaborate with cross-functional teams to ensure compatibility and scalability
  • Develop high-impact solutions to deliver to our large customer base
  • Participate in design discussions, code review, and communication with internal and external stakeholders
  • Work in a startup-like development environment, prioritizing innovation and customer needs
About the Team

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

We care about your career growth and strive to assign projects that help our team members develop their engineering expertise. Our senior members enjoy one-on-one mentoring and thorough code reviews.

What We Offer
  • A dynamic and innovative work environment
  • Opportunities for career growth and development
  • A collaborative and supportive team culture
  • Competitive compensation and benefits
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


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


  • Cupertino, California, United States Annapurna Labs Full time

    About the RoleWe are seeking a highly skilled TPM to lead our AWS AI Chips GTM efforts. As a key member of the Annapurna Labs team, you will be responsible for driving the adoption of our AI Chips across various industries and customer segments.Key ResponsibilitiesLead internal and external cross-team technical projects to accelerate the adoption of AWS AI...


  • Cupertino, California, United States Annapurna Labs Full time

    Join Our TeamAre you passionate about shaping the future of AWS AI Chips (AWS Inferentia/Trainium) Go to Market (GTM)? As a key member of the AWS AI Chips Business and GTM team, you will spearhead our most vital customer and industry partnership engagement initiatives.Your RoleIn this position, you will work closely with customers who deploy GenAI...


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


  • Cupertino, California, United States Amazon Full time

    Job SummaryWe are seeking a highly skilled Senior Software Development Engineer to join our team at Annapurna Labs, a subsidiary of Amazon. As a key member of our team, you will be responsible for designing, implementing, and optimizing firmware for our NVMe subsystem, DMA, and Crypto components.Key ResponsibilitiesDesign and implement firmware to support...


  • Cupertino, California, United States Amazon Full time

    Job SummaryWe are seeking a highly skilled Senior Software Development Engineer to join our team at Annapurna Labs, a subsidiary of Amazon. As a key member of our team, you will be responsible for designing, implementing, and optimizing firmware for our NVMe subsystem, DMA, and Crypto components.Key ResponsibilitiesDesign and implement firmware to support...


  • Cupertino, California, United States Annapurna Labs Full time

    About the RoleWe are seeking a highly skilled Physical Design Engineer to join our Cloud-Scale Machine Learning Acceleration team at Annapurna Labs. As a member of this team, you will be responsible for designing and optimizing hardware in our data centers, including AWS Inferentia and Trainium Systems.Key ResponsibilitiesCollaborate with RTL/logic designers...


  • Cupertino, California, United States Amazon Full time

    About the RoleWe are seeking a highly skilled Device Driver Engineer to join our team at Annapurna Labs, a part of Amazon Web Services (AWS). As a member of our team, you will be responsible for developing and maintaining the software drivers for our custom silicon chips, which power AWS's machine learning servers.Key ResponsibilitiesDesign and develop...


  • Cupertino, California, United States Amazon Full time

    About the RoleWe are seeking a highly skilled Device Driver Engineer to join our team at Annapurna Labs, a part of Amazon Web Services (AWS). As a member of our team, you will be responsible for developing and maintaining the software drivers for our custom silicon chips, which power AWS's machine learning servers.Key ResponsibilitiesDesign and develop...


  • Cupertino, California, United States Apple Full time

    Role OverviewWe are seeking a highly skilled Senior Software Engineer to join our dynamic team in enabling the next generation of intelligent experiences in Apple's products and services.Key ResponsibilitiesDesign, build, and maintain large-scale distributed systems to support the end-to-end machine learning (ML) lifecycle.Partner closely with data...


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

    About the RoleWe are seeking a highly skilled Senior Technical Program Manager to join our AWS AI Chips Go-to-Market team. As a key member of our team, you will be responsible for leading internal and external cross-team technical projects to accelerate adoption of AWS AI Chips.Key ResponsibilitiesLead complex customer and partner deliveriesDrive scale with...


  • 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

    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

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

    Join the Future of AWS AI ChipsAre you ready to shape the landscape of AWS AI Chips (AWS Inferentia/Trainium) Go to Market (GTM)? As a pivotal member of the global AWS AI Chips Business and GTM team, you will spearhead our most significant customer and industry partnership initiatives. Our clientele is at the forefront of deploying GenAI applications...


  • Cupertino, California, United States Amazon Full time

    About the RoleWe are seeking a highly skilled Senior Machine Learning Compiler Engineer III to join our Neuron Compiler team at Amazon. As a key member of this team, you will play a critical role in developing and scaling a compiler to enable the world's largest ML workloads to run performantly on custom Annapurna systems.Key ResponsibilitiesSolve complex...


  • 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

    Job SummaryWe are seeking an exceptional software engineer to expand and support Apple's machine learning framework (Core ML). As a key member of our team, you will be responsible for developing, maintaining, and supporting our ML frameworks and tools.Key ResponsibilitiesDevelop and maintain Apple's ML frameworks and toolsWork cross-functionally with...