The AWS Neuron team is seeking ML Compiler Engineers to optimize deep learning and GenAI workloads on AWS custom ML accelerators (Inferentia/Trainium). This role involves analyzing and optimizing system-level performance across the entire technology stack, from frameworks to runtime, and designing/implementing compiler optimizations. The position requires a passion for performance analysis, distributed systems, and machine learning, with a focus on improving the performance capabilities of the AWS Neuron SDK.

This role focuses on developing diverse datasets for training and evaluating AI models, utilizing methods like synthetic data generation, model-supported generation, and human-in-the-loop collections. The Language Engineer will collaborate with cross-functional teams to innovate and advance the state-of-the-art in AI model evaluation and training.

Senior Applied Science Manager to lead AI initiatives for seller growth at Amazon. The role involves developing scientific models for customer demand, selection identification, and prioritization signals. It also includes leading the development of next-generation agentic experiences for sellers, integrating insights into products like ABA and OX, and driving business impact through ML/LLM solutions.

Senior Systems Engineer for Amazon Prime Air's autonomous drone perception systems. This role involves translating operational requirements into system specifications, bridging ML algorithms with flight control, and ensuring perception systems meet aviation certification standards. The focus is on the integration and validation of ML-based perception for safe autonomous flight.

Software Development Manager to lead a team optimizing LLMs for inference on AWS custom accelerators (Neuron, Trainium, Inferentia). Focus on improving model enablement speed, experience, usability, and quality through features, infrastructure, tools, and automation. Requires strong background in LLM architectures, performance optimizations, and distributed inference.

Senior Machine Learning Engineer role at Amazon Prime Video focused on developing and launching AI solutions for recommendation and personalization systems. The role involves end-to-end ownership of ML models, including design, implementation, experimentation (A/B testing), and deployment for millions of customers. It requires experience with large-scale ML systems and recommendation systems.

Software Development Engineer focused on ML Systems within Amazon Annapurna Labs, working on AWS Neuron software for ML chips (Inferentia and Trainium). The role involves building and applying AI agents to accelerate customer adoption of this technology, optimizing performance, durability, cost, and security for AWS customers.

Senior ML Kernel Performance Engineer for AWS Neuron SDK, focusing on optimizing deep learning and GenAI workloads on custom ML accelerators (Inferentia, Trainium). The role involves designing and implementing high-performance compute kernels, optimizing performance at the hardware-software boundary, and collaborating with customers and internal teams on model enablement and acceleration.

The Annapurna Labs team at Amazon is seeking a Senior ML Kernel Performance Engineer to optimize deep learning and GenAI workloads on Amazon's custom ML accelerators (Inferentia and Trainium). This role involves crafting high-performance kernels, pushing the boundaries of AI acceleration at the hardware-software boundary, and collaborating with customers to enable their models. The engineer will work on compiler optimizations, performance analysis, and contribute to future architecture designs.

This role focuses on developing and launching end-to-end AI solutions for Prime Video's search and discovery systems, utilizing deep learning, GenAI, and reinforcement learning. The scientist will design and conduct experiments, collaborate with engineers and product managers, and publish research findings. The role is within the consumer domain, aiming to improve customer experience for millions of users.

Senior Machine Learning Compiler Engineer responsible for the ground-up development and scaling of a deep learning compiler stack for Amazon's ML accelerators (Inferentia and Trainium). The role involves architecting and implementing business-critical features, optimizing neural net models for custom hardware, and integrating with ML frameworks like PyTorch and TensorFlow.

Software Development Engineer III at Amazon Annapurna Labs, focusing on building and applying AI agents to simplify and accelerate customer adoption of AWS Neuron ML chips (Inferentia and Trainium). The role involves solving complex technical problems, designing and implementing innovative software solutions, and working with external and internal customers to identify adoption obstacles and opportunities in the Generative AI space.

This role is for a Sr. Machine Learning Compiler Engineer III on the AWS Neuron team, focusing on the development and scaling of a compiler for ML accelerators. The role involves architecting and implementing features for a deep learning compiler stack that optimizes neural network performance on custom AWS hardware, integrating with frameworks like PyTorch and TensorFlow. The goal is to provide significant performance improvements for large-scale ML workloads.

Senior Software Development Engineer to join Ring's AI Team, focusing on cloud services for machine learning operation pipelines that handle large-scale data and enable rapid model optimization. The role involves building and scaling platforms for AI model development and deployment, collaborating with cross-functional teams, and ensuring the delivery of robust backend systems.

Senior Software Development Engineer to join Ring's AI Team, focusing on cloud services for machine learning operation pipelines that handle large-scale data and enable rapid model optimization. The role involves building and scaling platforms for AI model development and deployment, collaborating with cross-functional teams, and ensuring the delivery of robust backend systems.

Sr. Applied Scientist role focused on developing AI prototypes and concepts for the JP Manga business, involving research, design, and training/tuning of NLP and Computer Vision models for applications like translation, summarization, extraction, boundary detection, image understanding, and generation. The role emphasizes tangible business impact and collaboration with product managers and engineers, with opportunities for publication.

Senior Software Development Engineer focused on Generative AI within Amazon's Annapurna Labs, specifically working with the Neuron SDK and ML chips (Inferentia and Trainium). The role involves building and applying AI agents to improve customer adoption of these chips, optimizing software solutions for performance, durability, cost, and security, and collaborating with cross-functional teams including compiler, hardware, and ML engineers. Experience in the Generative AI space is a hard requirement.

Data Scientist role focused on developing and implementing statistical, causal, and machine learning techniques for forecasting and inventory management within Amazon's retail supply chain. The role involves creating prototypes, collaborating with software teams for production implementation, and analyzing key business metrics to influence business direction.

Senior Applied Scientist role at Amazon Advertising focused on building and deploying end-to-end machine learning models to improve traffic monetization and merchandise sales. The role involves leading ML efforts, performing hands-on analysis, driving ambiguous projects, and establishing scalable processes for model development and deployment.

Software Development Engineer role on the AWS Solutions team, focusing on building and scaling the Machine Learning platform for Just Walk Out (JWO) Technology. The role involves developing algorithms for computer vision, image recognition, and machine learning within a distributed systems environment, with a focus on scaling ML platforms.

Senior Manager for AI Inference Technology, leading a team to build fundamental inference technology building blocks and libraries for AWS Neuron SDK, optimizing models for Trainium and Inferentia devices. Focuses on the full development life cycle of inference libraries, enabling customers to optimize LLMs, multimodal, and generative models.

Manager III leading a team of compiler engineers to develop, deploy, and scale a compiler targeting AWS Inferentia and Trainium ML accelerators. The role involves technical leadership, innovation, and collaboration with AWS ML services teams to ensure the Neuron SDK meets customer needs for high performance, low cost, and ease of use. Deep knowledge of resource management, scheduling, code generation, and optimization is required.

The role focuses on optimizing the performance of machine learning kernels for AWS's custom ML accelerators (Inferentia and Trainium) by developing and implementing high-performance compute kernels, optimizing compiler optimizations, and analyzing kernel-level performance. This involves working at the hardware-software boundary to ensure optimal performance for deep learning and GenAI workloads.

Senior Applied Scientist role at Amazon Prime Video focusing on Playback Intelligence. The role involves applying machine learning and data science to optimize video streaming quality, detect anomalies, and leverage LLMs and generative AI. Responsibilities include end-to-end ownership of product and user experience, translating business requirements into ML deliverables, defining research directions, conducting experiments, and mentoring junior scientists. The role requires experience in building ML models for business applications and designing AI solutions for real-world use cases.

Senior Software Development Engineer focused on pre-silicon hardware/software co-development for next-generation machine learning chips (like Trainium) used in AWS. The role involves working with architecture, design, and emulation teams, writing bare-metal software and ML workloads to verify chip functionality and performance.

Principal Applied Scientist role focused on applying machine learning, optimization, and economics to improve Amazon's Stores business, specifically in areas like delivery speed, seller fees, and LLM applications. The role involves leading a team, developing scientific models, benchmarks, and services, and deploying solutions in partnership with product teams.