AWS and Gray Lab at Johns Hopkins Whiting School of Engineering announce groundbreaking database for AI/ML antibody design

The Antibody Developability Benchmark is powered by one of the most diverse antibody datasets, enabling transparent performance evaluation for AI-guided antibody design.

Key takeaways
  • AWS and Johns Hopkins Engineering have launched the Antibody Developability Benchmark, a large-scale, diverse dataset for evaluating AI-guided antibody design.
  • The dataset includes 50 seed antibodies with four structural formats targeting 42 antigens, measuring six key developability traits.
  • It features engineered variants with both favorable and unfavorable developability outcomes, validated through wet-lab experiments.
  • The benchmark supports zero-shot learning, allowing models to be evaluated without prior exposure to the dataset, enhancing confidence in results.
  • The benchmark results are now available as part of Amazon Bio Discovery; additional benchmarks will be added over time and released in a paper later this year.
Was this answer helpful?

In 1986 the US Food and Drug Administration issued its first approval for human use of a therapeutic antibody. Despite steady advances in methodology, genetic sequencing, and biomedical science, 40 years later the process of discovering and optimizing therapeutic antibodies often remains prohibitively expensive, in terms of both cost and time. Recent experiences with pandemic-style infectious-disease outbreaks lend an even greater urgency to the need to more quickly and efficiently identify and develop these antibodies.

Artificial-intelligence- and machine-learning-guided approaches to antibody design, in the form of biological foundation models (BioFM), represent a significant opportunity to address these challenges. Models built using protein language models (pLMs) and structure-based deep-learning frameworks have significant potential to predict antibody developability properties — the characteristics that determine whether a molecule is manufacturable, stable, and safe as a therapeutic. The development of those tools could drastically shorten discovery timelines while also reducing experimental costs.

That potential, however, has been hindered by the lack of a public dataset that would allow researchers to benchmark those tools, a crucial step in the development of trustworthy in-silico tools for drug discovery. While there are existing public antibody datasets, they are too frequently limited by a focus on a single antibody format or target. Others are composed of naturally occurring or clinically advanced antibodies, a bias that severely limits their utility for training or evaluating predictive models.

“Trust in the predictions made by these models must be grounded in evaluations against experimental data that is sufficiently large and diverse,” explained Luca Giancardo, an applied scientist with Amazon Web Services (AWS) who works on the Amazon Bio Discovery team. “That data must be representative of the real sequence space encountered during antibody engineering and balanced in terms of developability outcomes.”

Jeffrey Gray is a professor in the Chemical and Biomolecular Engineering Department at the Johns Hopkins Whiting School of Engineering, where he leads the Gray Lab, which focuses on the computational prediction and design of protein structures. He is also the original developer of RosettaDock, a tool for the prediction of the structure of protein complexes from their constituent proteins.

Gray noted that while AI has made tremendous progress in the prediction and design of antibody properties, his own lab’s benchmarks have shown that current models do not yet reliably predict critical developability features, such as solubility and specificity, needed for efficient design of therapeutics. He cited the lack of diverse data in standardized conditions as a primary limitation for training models. That, coupled with the absence of a comprehensive, heterogenous, large-scale database, has acted as a significant drag on the potential of developing AI tools for antibody development.

Antibody developability benchmark

To that end, AWS, in collaboration with the Gray Lab and Johns Hopkins Engineering are announcing the launch of the Antibody Developability Benchmark, powered by the largest and most diverse antibody dataset in public literature. This is the first large-scale benchmark of antibody biophysical and biochemical properties designed to support the development and rigorous evaluation of in-silico antibody property predictors.

  • 0

    seed antibodies

  • 0

    structural formats

  • 0

    antigen targets

The Antibody Developability Benchmark is 20 times as diverse — in terms of antibody formats, targets, and developability profiles — as benchmarks currently available in the scientific literature. While other datasets may contain more individual antibody designs, they typically explore a single target or antibody framework with limited property coverage. The Antibody Developability Benchmark is unique in its combination of scale and heterogeneity, encompassing 50 seed antibodies, four structural formats, and 42 antigens. It also includes both favorable and unfavorable developability outcomes.

Gray lauded the opportunity to work with AWS experts, noting that the collaboration has enabled the creation of a dataset larger and more diverse than any of the publicly available datasets. He called the project an important next step toward fulfilling the promise of AI to improve human health.

AntibodyBenchmark-01-16x9.png
Existing public antibody datasets typically focus on a single target or format with limited property coverage (left). The Antibody Developability Benchmark is 20 times more diverse — spanning 50 seed antibodies, 4 structural formats, 42 antigens, and both favorable and unfavorable outcomes (right).

The Antibody Developability Benchmark includes the first heterogeneous antibody-property dataset explicitly designed to capture favorable and unfavorable developability profiles across multiple antigens and mutation strategies. Crucially, all data was affirmed via wet-lab experiments, providing ground truth validation that existing public benchmarks lack.

“This dataset will allow researchers to confidently be able to answer ‘Which model is better suited for our purposes?’,” noted Giancardo, whose Bio Discovery team led the development of the dataset. “Today there are many computational models coming out that are mostly evaluated on either proprietary data or public datasets, which are not representative of antibody heterogeneity. That means deciding what is better or worse is very, very hard — if not impossible.”

The unmatched diversity and deliberate heterogeneity of the Antibody Developability Benchmark will help make those determinations possible.

Michael Chungyoun, a PhD researcher at JHU who worked on the project, observed that the benchmark covers a wide space of antibodies, particularly in terms of their properties. He noted that allowing researchers to check against a very diverse benchmark can save time and labor by helping them compare models and choose the best approach.

The antibody dataset

The dataset consists of 50 clinically and scientifically relevant seed antibodies spanning four structural formats — IgG, VHH, NearGermline-IgG, and scFv — targeting 42 distinct antigens. It measures expression, purity, thermostability, aggregation, polyreactivity, and hydrophobicity — six traits that are essential in the development of viable therapeutic antibodies.

antibody structural format
The 50 seed antibodies in the Antibody Developability Benchmark span four structural formats: IgG (35), VHH (7), NearGermline-IgG (5), and scFv (3).

“The composition is a deliberate design choice,” Giancardo noted. “We strove to find a balance between heterogeneity of antibody classes, therapeutic targets, and mutation types, with the aim of creating benchmarks that would be generalizable across the structural diversity of the modern therapeutic-antibody landscape.”

Researchers at the Gray Lab, assisted by a sponsored research grant from AWS, helped select the seed antibodies for inclusion in the dataset. They were intentional about the seeds they chose, Chungyoun noted, opting in some cases for existing clinical-stage antibodies or FDA-approved antibodies. The team also selected antibodies more akin to those that circulate in the human body but aren't approved therapeutics. Those are called germline antibodies.

Chungyoun explained that germline antibodies are those found in the human body, and they have important biophysical characteristics. While some of those characteristics are shared with therapeutic antibodies, there are also differences between the two. The extent of those differences, and how to bridge that gap, is a vital and unanswered question.

Traditional antibody-based drug discovery begins with antibodies that come from animals or humans. Chungyoun explained that germline antibodies occasionally need to be modified to look more like therapeutics. That process is one researchers are still exploring.

Mutation strategy

The dataset also includes engineered variants of each seed antibody, generated by applying systematic mutation strategies to each seed.

“Initially, the hardest thing was essentially coming up with example sequences that would cover the broad spectrum of properties and the ways of mutating these sequences,” Giancardo explained. “It's challenging because you have to do it a priori until you do it, and then you don't know what will come out.”

Working with Johns Hopkins Engineering, Giancardo and his team systematically engineered variants employing a variety of approaches, including protein-language-model-guided (pLM-guided) versus non-pLM-guided mutation selection and amino acid substitutions versus insertions/deletions.

“Protein language models are essentially the equivalent of large language models [LLMs] for the protein world,” Giancardo said. “There are multiple ways of looking at proteins. A common way is expressing them as a string of amino acids, which are essentially letters.” When some of the letters in the amino acid chains are masked, the models can be trained to fill in the gaps — the same "self-supervised" approach used to train LLMs. The models can also be trained to predict what changes inserting a different letter or letters — i.e., mutation — will yield. That approach resulted in a wide variety of mutations — up to 99 engineered variants per seed.

The breadth and depth of those mutations contribute to another distinguishing feature of the Antibody Developability Benchmark: its deliberate heterogeneity. The inclusion of both favorable, or developable, and unfavorable, or poorly developable, examples sets it apart from existing datasets.

“This range is essential for training and evaluating machine learning models, which require balanced label distributions and exposure to the failure modes they are intended to predict and avoid,” Giancardo explained. He also clarified that those failures still fall within a range of viability.

“These are not examples that are obviously wrong but rather bad examples that have a fighting chance," he added. "These all still meet some baseline quality assessment, meaning researchers could reasonably send them to a wet-lab partner to test.”

Zero-shot learning

Gray and his team at Hopkins Engineering also collaborated with their AWS counterparts by selecting and running existing open-source antibody design and prediction models on their own. They then shared their findings with the Bio Discovery team, who compared the results those models generated against the benchmarking dataset without exposing those models to the information in that dataset.

“This is essentially zero-shot inference,” Giancardo said. That siloed approach allowed both sides to have greater confidence in the results the Antibody Developability Benchmark generated. “The fact that we operated separately gave us confidence that we were not introducing errors. There is no data leakage of any sort, even from an external perspective.”

The teams compared their data and used those results to further fine-tune the Antibody Developability Benchmark. That iterative process means researchers who utilize the benchmark can have greater confidence about the viability of their models before the necessary, and costly, step of working with a wet lab partner. That can also shorten the overall timeline in terms of experimentation.

“When you are confident enough to do a screen, then you can turn to the wet lab, get new metrics, and further train on those results, which will be much, much, much more meaningful,” Giancardo explained.

The future

Researchers at both AWS and Hopkins Engineering emphasized the importance of sharing model benchmarks based on the Antibody Developability Benchmark Dataset with the larger scientific community. The benchmark results are now available as part of Amazon Bio Discovery; additional benchmarks will be added over time and released in a paper later this year.

The sharp uptick in proposed protein AI models has researchers excited, but the expense and time commitment of wet labs has meant researchers have thus far been unable to compare those models head to head, Chungyoun observed. He noted that the launch of this dataset means those researchers now have an opportunity to learn which model properties improve performance. That can serve to illuminate the connection between what models learn and how those models can be improved to better predict those properties.

The dataset won’t remain static either: more models and properties will be added in the future.

"The database has the potential to surface models and tools that may have previously gone unrecognized — research published in lesser-known venues or work that simply didn't receive the attention it deserved," said Nina Cheng, a senior science manager in the AWS Life Sciences organization. "This database can play a key role in bringing that kind of overlooked work to light."

Acknowledgements

Amazon Bio Discovery Science and product team: Luca Giancardo, Yue Zhao, Melih Yilmaz, Kemal Sonmez, Lan Guo, Gordon Trang, Edward Lee, Chuanyui Teh, Fangda Xu, Nina Cheng, Jiwon Kim.

Research areas

Related content

US, WA, Seattle
The People eXperience Technology (PXT) Central Science (PXTCS)'s mission is to make PXT the most scientific, technologically proficient, and inclusive HR organization in the world. PXTCS does this by accelerating scientific rigor in business-led initiatives, working backwards from employee experience, and delivering science-driven products that improve the well-being of and value of work for Amazonian worldwide. We are seeking an Applied Scientist to build production machine learning systems that turn Amazon's employee listening signals into insight at scale. You will design, develop, and deploy ML solutions that transform large volumes of employee feedback — survey responses, free-text comments, and other listening channels — into signals that drive strategic decision-making across the organization. In this role, you will work on challenging problems spanning natural language processing, large language models, predictive modeling, and generative AI applications applied to employee experience. You will collaborate with cross-functional teams to translate ambiguous business challenges into rigorous technical solutions. Key job responsibilities - Apply and adapt state-of-the-art NLP and machine learning techniques to extract structured, reliable signals from employee listening data (surveys, free-text feedback, and related channels), using reasonable assumptions, data, and customer requirements. - Design, develop, and implement small-to-medium ML components on top of the employee listening system — such as theme and sentiment extraction, text classification, and aggregated signal generation — with input and guidance from senior scientists, taking ownership of the code in your components. - Write secure, stable, testable, maintainable, well-reviewed code (at the SDE I bar) to deliver solutions into production that benefit customers and the business. - Design features and signals responsibly, applying privacy-preserving practices appropriate to sensitive employee data. - Collaborate with cross-functional partners to understand business context and impact, and help mentor interns. - Communicate complex technical concepts to diverse audiences, from technical peers to senior leadership. About the team The People eXperience and Technology Central Science Team (PXTCS) uses economics, applied science, statistics, and machine learning to proactively identify mechanisms and process improvements which simultaneously improve Amazon and the lives, wellbeing, and the value of work to Amazonians. We build and apply science on top of Amazon's employee listening systems — turning what Amazonians tell us into products and insights that improve their experience. We are an interdisciplinary team that combines the talents of science and engineering to develop and deliver solutions that measurably achieve this goal.
US, CA, Sunnyvale
The Frontier AI Assets (FAI) team is looking for a passionate, talented, and inventive Principal Applied Scientist with a strong deep learning and reinforcement learning background to build the training and evaluation assets that advance frontier Generative AI (GenAI) with Large Language Models (LLMs). AI models become more capable largely because of the learning environments, the RL gyms, in which they are trained. FAI builds those gyms and the platform that certifies them, so that model training teams across Amazon can rely on high-quality, graded assets. As a Principal on this team, you will set the technical direction for how these gyms and the tooling around them are built, evaluated, and scaled. Key job responsibilities As a Principal Applied Scientist with FAI, you will: •Set technical direction across data collection, sampling, and evaluation for RL gyms focused on agentic tool use across a range of real-world domains. •Design and lead experiments that produce reliable, efficient signal on what drives model improvement. •Build gym-building tooling and automated quality control that scale with demand. •Advance the science of how gyms are evaluated and validated before they're used in model training. •Act as a hands-on technical leader: produce exemplary, high-quality assets yourself, and audit and support more junior scientists and engineers. •Partner closely with domain experts and cross-functional teams, turning real-world problems into challenging, high-value tasks. •Lead scientific reviews, scrutinize experimental design and verification, probe assumptions, and align teams toward coherent strategies. Contribute to hiring, mentorship, and the development of other scientists. About the team The FAI team has the mission to push the envelope in GenAI with LLMs, in order to provide the best-possible experience for our customers. We build frontier training and evaluation assets, RL gyms and the certification platform behind them, that directly shape how Amazon's models are trained and improved. We value scientific depth, hands-on building, and turning research into measurable impact at scale.
US, CA, Sunnyvale
Amazon Robotics is seeking an exceptional Applied Scientist to join our Foundation Models and Agentic AI team. This role presents an opportunity to shape the future of robotics through innovative applications of large vision-language-action (VLA) models, reasoning and task planning models, and reinforcement learning. At Amazon Robotics, we leverage advanced robotics, machine learning, and artificial intelligence to solve complex operational challenges at unprecedented scale. Our fleet of robots operates across hundreds of facilities worldwide, working in sophisticated coordination to fulfill our mission of customer excellence. We are pioneering the development of robotics foundation models that: - Enable unprecedented generalization across diverse tasks - Integrate multi-modal learning capabilities (visual, depth, force, tactile, linguistic) - Accelerate policy skill acquisition through imitation learning and reinforcement learning - Enhance robotic perception and environmental understanding - Streamline development processes through reusable capabilities The ideal candidate will contribute to research that bridges the gap between theoretical advancement and practical implementation in robotics. You will be part of a team that's revolutionizing how robots learn, adapt, and interact with their environment. Join us in building the next generation of intelligent robotics systems that will transform the future of automation and human-robot collaboration. Key job responsibilities As an Applied Scientist in the Applied AI and Robotics Foundations Model team, you will: - Model Development and Training: Designing and implementing the model architectures, training and fine tuning the foundation models using various datasets, and optimize the model performance through iterative experiments - Data Management: Process and prepare training data, including data governance, provenance tracking, data quality checks and creating reusable data pipelines. - Experimentation and Validation: Design and execute experiments to test model capabilities on the simulator and on the embodiment, validate performance across different scenarios, create a baseline and iteratively improve model performance. - Code Development: Write clean, maintainable, well commented and documented code, contribute to training infrastructure, create tools for model evaluation and testing, and implement necessary APIs - Research: Stay current with latest developments in foundation models and robotics, assist in literature reviews and research documentation, prepare technical reports and presentations, and contribute to research discussions and brainstorming sessions. - Collaboration: Work closely with senior scientists, engineers, and leaders across multiple teams, participate in knowledge sharing, support integration efforts with robotics hardware teams, and help document best practices and methodologies. A day in the life Amazon offers a full range of benefits that support you and eligible family members, including domestic partners and their children. Benefits can vary by location, the number of regularly scheduled hours you work, length of employment, and job status such as seasonal or temporary employment. The benefits that generally apply to regular, full-time employees include: - Medical, Dental, and Vision Coverage - Maternity and Parental Leave Options - Paid Time Off (PTO) - 401(k) Plan If you are not sure that every qualification on the list above describes you exactly, we'd still love to hear from you! At Amazon, we value people with unique backgrounds, experiences, and skillsets. If you’re passionate about this role and want to make an impact on a global scale, please apply! About the team Are you inspired by invention? Is problem solving through teamwork in your DNA? Do you like the idea of seeing how your work impacts the bigger picture? Answer yes to any of these and you’ll fit right in here at Amazon Robotics. We are a smart, collaborative team of enthusiastic doers that work passionately to apply innovative advances in robotics and software to solve real-world challenges that will transform our customers’ experiences in ways we can’t even image yet. We invent new improvements every day. We are Amazon Robotics and we will give you the tools and support you need to invent with us in ways that are rewarding, fulfilling and fun.
IN, KA, Bengaluru
RBS (Retail Business Services) Tech team works towards enhancing the customer experience (CX) and their trust in product data by providing technologies to find and fix Amazon CX defects at scale. Our platforms help in improving the CX in all phases of customer journey, including selection, discoverability & fulfilment, buying experience and post-buying experience (product quality and customer returns). The team also develops GenAI platforms for automation of Amazon Stores Operations. As a Sciences team in RBS Tech, we focus on foundational ML research and develop scalable state-of-the-art ML solutions to solve the problems covering customer experience (CX) and Selling partner experience (SPX). We work to solve problems related to multi-modal understanding (text and images), task automation through multi-modal LLM Agents, supervised and unsupervised techniques, multi-task learning, multi-label classification, aspect and topic extraction for Customer Anecdote Mining, image and text similarity and retrieval using NLP and Computer Vision for product groupings and identifying duplicate listings in product search results Key job responsibilities As a Senior Research Scientist, you will be responsible to design and deploy scalable GenAI, NLP and Computer Vision solutions that will impact the content visible to millions of customer and solve key customer experience issues. You will develop novel LLM, deep learning and statistical techniques for task automation, text processing, image processing, pattern recognition, and anomaly detection problems. You will define the research and experiments strategy with an iterative execution approach to develop AI/ML models and progressively improve the results over time. You will partner with business and engineering teams to identify and solve large and significantly complex problems that require scientific innovation. You will help the team leverage your expertise, by coaching and mentoring. You will contribute to the professional development of colleagues, improving their technical knowledge and the engineering practices. You will independently as well as guide team to file for patents and/or publish research work where opportunities arise. The RBS org deals with problems that are directly related to the selling partners and end customers and the ML team drives resolution to organization level problems. Therefore, the Senior Research Scientist role will impact the large product strategy, identifies new business opportunities and provides strategic direction which is very exciting.
IN, HR, Gurugram
Lead ML teams building large-scale forecasting and optimization systems that power Amazon’s global transportation network and directly impact customer experience and cost. As an Sr Applied Scientist, you will set scientific direction, mentor applied scientists, and partner with engineering and product leaders to deliver production-grade ML solutions at massive scale. Key job responsibilities 1. Lead and grow a high-performing team of Applied Scientists, providing technical guidance, mentorship, and career development. 2. Define and own the scientific vision and roadmap for ML solutions powering large-scale transportation planning and execution. 3. Guide model and system design across a range of techniques, including tree-based models, deep learning (LSTMs, transformers), LLMs, and reinforcement learning. 4. Ensure models are production-ready, scalable, and robust through close partnership with stakeholders. Partner with Product, Operations, and Engineering leaders to enable proactive decision-making and corrective actions. 5. Own end-to-end business metrics, directly influencing customer experience, cost optimization, and network reliability. 6. Help contribute to the broader ML community through publications, conference submissions, and internal knowledge sharing. A day in the life Your day includes reviewing model performance and business metrics, guiding technical design and experimentation, mentoring scientists, and driving roadmap execution. You’ll balance near-term delivery with long-term innovation while ensuring solutions are robust, interpretable, and scalable. Ultimately, your work helps improve delivery reliability, reduce costs, and enhance the customer experience at massive scale.
US, WA, Bellevue
Have you ever ordered a product on Amazon and when that box with the smile arrived you wondered how it got to you so fast? Have you wondered where it came from and how much it cost Amazon to deliver it to you? Have you also wondered what are different ways that the transportation assets can be used to delight the customer even more. If so, the Amazon transportation Services, North America Sort Center Science team is for you . We manage the delivery of tens of millions of products every week to Amazon’s customers, achieving on-time delivery in a cost-effective manner. We are looking for an enthusiastic, customer obsessed Applied Scientist with strong scientific thinking, good software and statistics experience, skills to help manage projects and operations, improve metrics, and develop scalable processes and tools. The primary role of an Applied Scientist within Amazon is to address business challenges through building a compelling case, and using data to influence change across the organization. This individual will be given responsibility on their first day to own those business challenges and the autonomy to think strategically and make data driven decisions. Decisions and tools made in this role will have significant impact to the customer experience, as it will have a major impact on how we operate the middle mile network. Ideal candidates will be a high potential, strategic and analytic graduate with a PhD in (Operations Research, Statistics, Engineering, and Supply Chain) ready for challenging opportunities in the core of our world class operations space. Great candidates have a history of operations research, machine learning , and the ability to use data and research to make changes. This role requires robust skills in research and implementation of scalable products and models . This individual will need to be able to work with a team, but also be comfortable making decisions independently, in what is often times an ambiguous environment. Key job responsibilities Key job responsibilities - Design and develop advanced mathematical optimization and machine learning solutions in the domains of network design and optimization. - Research, prototype, simulate, and experiment with these models using programming languages such as Java and Python; participate in the production level deployment. - Create, enhance, and maintain technical documentation and science designs. - Present to other Scientists, Product, and Software Engineering teams, as well as Stakeholders. - Lead project plans from a scientific perspective by managing product features, technical risks, milestones and launch plans.
US, WA, Seattle
Prime Video is a first-stop entertainment destination offering customers a vast collection of premium programming in one app available across thousands of devices. Prime members can customize their viewing experience and find their favorite movies, series, documentaries, and live sports – including Amazon MGM Studios-produced series and movies; licensed fan favorites; and programming from Prime Video subscriptions such as Apple TV+, HBO Max, Peacock, Crunchyroll and MGM+. All customers, regardless of whether they have a Prime membership or not, can rent or buy titles via the Prime Video Store, and can enjoy even more content for free with ads. Are you interested in shaping the future of entertainment? Prime Video's technology teams are creating best-in-class digital video experience. As a Prime Video team member, you’ll have end-to-end ownership of the product, user experience, design, and technology required to deliver state-of-the-art experiences for our customers. You’ll get to work on projects that are fast-paced, challenging, and varied. You’ll also be able to experiment with new possibilities, take risks, and collaborate with remarkable people. We’ll look for you to bring your diverse perspectives, ideas, and skill-sets to make Prime Video even better for our customers. With global opportunities for talented technologists, you can decide where a career Prime Video Tech takes you! Key job responsibilities The science challenges include: - Heterogeneous customer responses. The same customer responds differently to ads depending on what they’re watching, how engaged they are, and their broader streaming context. We need to understand and predict this variation to make better advertising decisions in real time. - Signal sparsity in streaming. Unlike search or retail advertising, streaming offers no clicks, no conversions, and limited direct feedback. We must develop creative approaches to infer customer preferences, intent, and tolerance from indirect behavioral signals. - Personalization at scale. A one-size-fits-all ad experience leaves value on the table for both customers and the business. We build systems that adapt ad load, placement, and content to individual viewers across 100M+ customers and 100k+ titles. - Small effects, large variance. Ad interventions typically produce 0.1-2% shifts in engagement, effects easily overwhelmed by natural behavioral variance. Measuring, attributing, and optimizing these small signals requires rigorous experimental design and causal methodologies. - Competing objectives. Revenue, customer engagement, long-term retention, and advertiser value are in tension. We develop principled frameworks to navigate these tradeoffs and optimize for sustainable outcomes rather than any single metric. A day in the life - Contribute to the research and development of ML models that personalize advertising decisions for 100M+ customers across 100k+ titles, with production deployment in mind - Develop deep learning architectures (multi-task learning, embedding-based representations) for customer behavior prediction at scale - Design and analyze large-scale A/B experiments, applying causal inference techniques to measure and optimize the impact of ad strategies on customer engagement and monetization - Partner with engineering to ensure models meet production latency and scalability requirements - Collaborate with product managers to frame business problems as tractable ML problems and translate findings into product decisions - Shape the team’s scientific roadmap, identifying high-impact research directions About the team Prime Video Ads is pioneering the use of Generative AI, Deep Learning and Causal ML to empower the next generation of Ads experience in Prime Video. Our vision is to build the world’s most valuable and differentiated ad-supported streaming platform: one where every ad is relevant to the viewer, and every dollar spent by an advertiser delivers measurable results. We are seeking lead applied scientists who have demonstrated experience in advancing state of the art ML algorithms, particularly in Personalization, Relevance, Casual Inference and Gen-AI areas. Your role will be to deliver these innovations as production-ready systems at Amazon scale.
US, WA, Seattle
Pricing is one of the most consequential decisions Amazon makes, and the science behind it needs to be causally rigorous, not just predictive. The P2 Optimization Science team builds the machine learning (ML) systems that power Amazon's pricing decisions at scale: demand lift models, customer lifetime value (LTV) frameworks, and the experimentation infrastructure that validates whether pricing changes actually work. We're looking for a Research Scientist to own causal inference at the intersection of ML and pricing experimentation. Our team has identified a real gap: the methodological bridge between econometric analysis and production-scale ML pipelines needs a practitioner who lives in both worlds. You'll build conditional average treatment effect (CATE) estimation models, design analysis workflows for pricing experiments, and develop reusable causal ML infrastructure that the broader team — including non-ML scientists — can rely on. If you're a scientist who wants to tackle hard causal identification problems in a high-stakes production environment, and who finds satisfaction in making rigorous methods accessible to a broader team, this role is for you. Key job responsibilities - Design, train, evaluate, and deploy end-to-end causal ML pipelines for pricing use cases, including CATE estimation models and heterogeneous treatment effect frameworks. - Serve as the team's causal ML methodology expert — defining identification strategies, model selection criteria, evaluation standards, and the tradeoffs between econometric and ML approaches to causal estimation. - Contribute causal analysis methodology to pricing A/B test post-analysis and build reusable tooling that economists and other non-ML scientists can adopt without requiring deep ML expertise. - Assess emerging causal inference methods (e.g., synthetic difference-in-differences, generalized random forests, causal representation learning) for applicability to Amazon's pricing context, and author internal methodology proposals for adoption. - Partner with senior economists on identification strategy and causal assumptions, collaborate with software and data engineering partners on production deployment, and align with product managers on experiment design requirements. A day in the life Your work directly shapes the prices customers see on hundreds of millions of Amazon products. In a given week you might investigate an optimization anomaly in simulation and trace it to an unmodeled market dynamic, then design an offline evaluation framework to benchmark competing approaches before committing to online testing. You'll collaborate with senior economists on identification strategy for a pricing experiment, present a science proposal for incorporating a new competitiveness signal into an optimization system, or work with the experimentation platform team on randomization design. You'll also have opportunities to write up novel scientific findings for submission to venues such as KDD, NeurIPS, or the ACM Conference on Economics and Computation. About the team The P2 Optimization Science team sits at the center of Amazon's pricing decisions. We build the ML systems, causal inference pipelines, and experimentation infrastructure that determine how prices are set across Amazon's global catalog. Our team brings together research scientists, economists, and engineers who collaborate closely to turn rigorous methodology into production-quality systems with measurable business impact. We value scientific rigor, clear communication across disciplines, and a culture where every team member's work is designed to be extensible and useful beyond their own project. If you want your science to directly influence how hundreds of millions of customers experience Amazon, this is the team to join.
US, CA, San Francisco
Amazon Music is an immersive audio entertainment service that deepens connections between fans, artists, and creators. From personalized music playlists to exclusive podcasts, concert livestreams to artist merch, Amazon Music is innovating at some of the most exciting intersections of music and culture. We offer experiences that serve all listeners with our different tiers of service: Prime members get access to all the music in shuffle mode, and top ad-free podcasts, included with their membership; customers can upgrade to Amazon Music Unlimited for unlimited, on-demand access to 100 million songs, including millions in HD, Ultra HD, and spatial audio; and anyone can listen for free by downloading the Amazon Music app or via Alexa-enabled devices. Join us for the opportunity to influence how Amazon Music engages fans, artists, and creators on a global scale. The Data, Insights, Science and Optimization, Consumer Product and Tech (DISCO CPT) team is looking for a Data Scientist to join a team of Data Scientists, Analytics Leads, Business Intelligence Engineers and Data Engineers who analyze big data, provide analytics and insights as well as build models and algorithms that power product for personalized experiences. The team focuses on accelerating Amazon Music customer growth by empowering Product teams to make sound, customer-centric decisions through data and insights. We build data pipelines, science models, self-service analytics, insights and dashboards, deliver engagement and retention at scale. In this role, you will pioneer and execute advanced analytics and experimentation efforts for the Amazon Music product team. You will work with stakeholders and partners to deliver experimentation insights, author scientific artifacts, build scalable models to advance our analytics and science products. The ideal candidate must be willing to effectively lead science solutions, prioritize across multiple stakeholders and projects and be ready to jump into a fast-paced, dynamic and fun environment. Key job responsibilities 1. Data Scientist I on this role will be a thought partner to Product and Engineering organizations through their expertise in experimentation, statistical modeling and advanced analytics 2. Conceptualize and drive strategic growth areas: app performance, stablity and latency experimentation, Gen AI initiatives, and product excellence. 3. Bridge science, analytics, and engineering to deliver solutions informing strategic decisions on AM technology investments and long-term business performance 4. Drive experimentation, measurement efficiency tooling, and ROI tracking for AM leadership 5. Build and maintain causal models to understand customer sentiment and its impact on key engagement/retention metrics 6. Develop structural and predictive models using data science workflows to deliver measurable customer results A day in the life 1. Lead strategic initiatives across visual and voice experiences, experimentation and customer retention stakeholders 2. Conduct experimentation, build causal models to better understand key drivers for customer retention 3. Evaluate existing metrics and lead deep dives to ensure the right measurement framework is leveraged to measure and enhance user adoption through visual and voice interfaces 4. Partner with product managers to translate business questions into actionable science projects 5. Influence product roadmaps through data-driven insights on customer behavior patterns 6. Identify and resolve customer experience friction points with cross-functional teams Drive adoption of best practices and innovative solutions across the organization
US, NY, New York
The Measurement Science team in the Measurement, Ad Tech, and Data Science (MADS) organization of Amazon Ads serves a centralized role developing solutions for a multitude of performance measurement products. We create solutions which measure the comprehensive impact of their ad spend, including sales impacts both online and offline and across timescales, and provide actionable insights that enable our advertisers to optimize their media portfolios. We leverage a host of scientific technologies to accomplish this mission, including Generative AI, classical ML, Causal Inference, Natural Language Processing, and Computer Vision. This Principal Applied Scientist leads the technical vision and execution for Amazon Advertising's performance measurement. Our measurement products and tools help customers understand the holistic impact of their spend on Amazon Ads and optimize their portfolios. This is a complex, multi-disciplinary scientific space spanning machine learning, causal inference, statistics and experimentation, natural language processing, and generative AI. As a GenAI-first organization, we build foundational and agentic models that power advertiser use cases across Ads, while empowering our Applied Scientists to directly build and ship products. You will guide a team of scientists with expertise across different domains to come together and innovate on some of the hardest problems in advertising, made more challenging by evolving privacy regulations and technological changes. In measuring the comprehensive impact of advertising, key challenges you will tackle include capturing impact: 1) both for identified and increasingly unidentified web traffic in light of an evolving global privacy landscape; 2) for outcomes across retailers, including both within and beyond Amazon; and 3) for both short- and long-term impacts of advertising. Key job responsibilities * Driving the scientific vision of the teams in your organization and you advise and influence its technical leadership on performance measurement models and products. * Identifying, tackling, and proposing innovative solutions to intrinsically hard, previously unsolved problems. You solve problems pragmatically, applying judgment and experience to balance trade-offs between competing interests. * Bringing clarity to complex problems, probing assumptions, illuminating pitfalls, fostering shared understanding, and guiding towards effective solutions. * Serving and being recognized by internal and external peers as a thought leader in modeled measurement in Ads. You will engage routinely with the broader scientific community, e.g. as a reviewer, program committee member, workshop organizer, or chair at peer-reviewed internal/external conferences. * Leading by example, with your papers, analyses, and code setting the standard in your organization for scientific engineering excellence. * Leading science and design reviews, aligning teams across your organization around coherent algorithmic strategies. * Influencing your team’s science and business strategy by driving one or more team roadmaps contributing to the organization’s roadmap and taking responsibility for some organizational goals. You drive multiple new product features from inception to production launch. * Guiding the career development of others, actively mentoring and educating the larger applied science community on trends, technologies, and best practices. You drive the documentation of scientific innovation. You attract bar-raising talent from academia and industry. * Staying abreast of academic and industry trends, knowing the state of the art, and building on top of existing solutions instead of reinventing when possible. About the team We are a team of scientists across Applied, Research, Data Science and Economist disciplines. You will work with colleagues with deep expertise in ML, NLP, CV, Gen AI, and Causal Inference with a diverse range of backgrounds. We partner closely with top-notch engineers, product managers, sales leaders, and other scientists with expertise in the ads industry and on building scalable modeling and software solutions.