Ph3Overview /h3 pThe Annapurna Labs team at Amazon Web Services (AWS) builds AWS Neuron, the software development kit used to accelerate deep learning and GenAI workloads on Amazon’s custom machine learning accelerators, Inferentia and Trainium. The Acceleration Kernel Library team focuses on maximizing performance for AWS's custom ML accelerators. Engineers here craft high‑performance kernels for ML functions, combining deep hardware knowledge with ML expertise to push the boundaries of AI acceleration. The AWS Neuron SDK includes an ML compiler, runtime, and application framework that integrates with popular ML frameworks like PyTorch to deliver high performance for inference and training on AWS accelerators. This role sits at the intersection of machine learning, high‑performance computing, and distributed architectures, helping shape the future of AI acceleration technology. This is an opportunity to work on cutting‑edge products, architect and implement business‑critical features, publish cutting‑edge research, and mentor a team of engineers in a fast, innovative culture. The team collaborates closely with customers on model enablement, providing direct support and optimization expertise to ensure workloads run optimally on AWS ML accelerators. /p pExplore the product and our history: /p h3Key job responsibilities /h3 ul liDesign and implement high-performance compute kernels for ML operations, leveraging the Neuron architecture and programming models. /li liAnalyze and optimize kernel-level performance across multiple generations of Neuron hardware. /li liConduct detailed performance analysis using profiling tools to identify and resolve bottlenecks. /li liImplement compiler optimizations such as fusion, sharding, tiling, and scheduling. /li liWork directly with customers to enable and optimize their ML models on AWS accelerators. /li liCollaborate across teams to develop innovative kernel optimization techniques. /li /ul h3A day in the life /h3 ul liBuild high-impact solutions to deliver to our large customer base. /li liParticipate in design discussions, code review, and communicate with internal and external stakeholders. /li liWork cross‑functionally to help drive business decisions with your technical input. /li liWork in a startup‑like development environment, focusing on the most important tasks. /li /ul h3About the team /h3 ul liDiverse Experiences: AWS values diverse experiences. We encourage candidates to apply even if you do not meet all listed qualifications. /li liWhy AWS: AWS is the world’s most comprehensive and broadly adopted cloud platform. We pioneered cloud computing and continue to innovate. /li liInclusive Team Culture: We embrace differences and maintain an inclusive culture with multiple affinity groups and ongoing learning experiences, guided by AWS Leadership Principles. /li liWork / Life Balance: We value balance and offer flexible working hours. /li liMentorship Career Growth: We support new members with mentorship and project assignments that foster growth. /li /ul h3Basic Qualifications /h3 ul li3+ years of non‑internship professional software development experience. /li li2+ years of non‑internship design or architecture experience (design patterns, reliability and scaling) of new and existing systems. /li liExperience programming with at least one software programming language. /li /ul h3Preferred Qualifications /h3 ul li3+ years of full SDLC experience including coding standards, code reviews, source control, build processes, testing, and operations. /li liBachelor’s degree in computer science or equivalent. /li /ul pLos Angeles County applicants: job duties include working safely, communicating effectively, and following laws and policies. Criminal history may affect employment opportunities. We will consider qualified applicants with arrest and conviction records per the Los Angeles County Fair Chance Ordinance. /p pOur inclusive culture empowers Amazonians to deliver the best results for customers. If the country / region you’re applying in isn’t listed, contact your Recruiting Partner. /p pAmazon is an equal‑opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status. /p pCompensation reflects labor costs across US markets. The base pay for this position ranges from $129,300 / year to $223,600 / year, depending on location and experience. Amazon is a total compensation company with potential equity, sign‑on, and other benefits. For more information, visit /p /p #J-18808-Ljbffr