Advancing AI for biology: Teaching models to design and characterize antibodies

Three new papers from Amazon Bio Discovery address bottlenecks in AI-driven antibody engineering, from benchmarking binding predictors to experimentally validating de novo design.

Key takeaways
  • Amazon Bio Discovery has developed three AI approaches to accelerating antibody drug design: MochiBind for ranking binding candidates by sequence alone, CA-MAP for predicting antibody developability properties while accounting for lab-specific batch effects, and an agent-guided design system validated experimentally on a novel cancer target.
  • MochiBind reframes binding affinity prediction as relative ranking rather than absolute values, achieving 10% higher accuracy than structure-based competitors while enabling 100-fold faster inference, scoring 200,000 antibody pairs in 13 seconds on a CPU.
  • CA-MAP addresses batch effects through context-aware learning, maintaining 0.99 correlation accuracy even with simulated batch offsets where standard models degrade to 0.58, while using 1/200th as many parameters and running 200x faster than competing large language models.
  • The end-to-end design campaign successfully generated 46 validated strong binders against a desmoplastic small round cell tumor target with no prior structural or antibody information.
Was this answer helpful?

Monoclonal antibodies are one of the workhorses of biopharmaceutical development, with over 100 FDA-approved drugs and well-established manufacturing, regulatory, and clinical-development pathways. Yet conventional antibody discovery remains hampered by mounting costs and long timelines, typically six to twelve months to get from a target to a lead candidate.

By designing and characterizing therapeutic antibodies computationally, AI promises to make development cheaper, faster, and more flexible. But scientific questions abound.

Development of an antibody-based drug hinges on three factors: the best binding site on the target, which candidates bind to it most tightly, and whether any of them can survive manufacturing and the clinic. For each, the field has predictive models that do well on familiar targets and assays but considerably worse on unfamiliar ones. Benchmarks built around in-distribution accuracy have made that gap difficult to measure — and to close.

Three papers from our science team at Amazon Bio Discovery, an AI-powered application that gives scientists access to biological AI models and integrated lab services to design and test novel drug candidates, tackle research questions about each of these three factors. Two are peer-reviewed journal papers on prediction: ranking candidates by binding strength and flexibly predicting developability. The third brings prediction into an end-to-end design process, navigates the selection of binding sites with an agent, and delivers experimentally validated antibody hits against a novel cancer target.

AI searches a vast molecular landscape to find the most promising target for antibody design.

Ranking binders from sequence alone

One of the biggest questions in antibody design is which candidates bind the best. In "A systematic evaluation framework for universal antibody-antigen binding affinity prediction and candidate recommendation", published in iScience, we propose a new framework to assess binding affinity predictors and train a new sequence-based predictor, MochiBind.

Most affinity predictors are evaluated on their ability to predict the absolute binding affinity, on antigens that appear in their training data, against test sets that contain few or no nonbinders. Each of these characteristics makes the evaluation easier than the intended application. Absolute affinity values are not comparable across assays, and performance degrades for antigens the model has not seen. The practical use case, meanwhile, involves ranking a pool of thousands of candidates, most of which don’t bind to the target at all, to pick the ones worth testing in the lab. Surveying seven prior studies, we found that none satisfied all the conditions necessary to train a reliable universal predictor.

We therefore reframed the task. Rather than predicting an absolute number, MochiBind predicts which of two antibodies against the same antigen binds more tightly. We begin by using a pretrained protein language model (ESM-2) to embed residues of antibody-antigen complexes in a representational space. We then compute the mean of each complex’s residue embeddings, to give it a single embedding.

A specially trained network layer projects these embeddings into a lower-dimensional space, and predicts relative binding strength from the difference between the two projections.[HL2]  Pairwise comparisons are then aggregated into a global ranking over the candidate pool using TrueSkill, a Bayesian rating algorithm originally developed for ranking video game players based on match outcomes. No structural input is required at any stage.

This formulation has two practical advantages: relative orderings are more consistent across assays than absolute values, so the training signal is less sensitive to measurement noise, and the output is the ranked list the discovery process needs.

Our paper also presents a novel evaluation framework. We used the AlphaBind dataset, which covers four antigen systems (targeting TIGIT, PD-1, HER2, and theSARS-CoV-1 RBD) with roughly 30,000 experimentally characterized variants for each and pairwise sequence similarity between antigens that’s close to zero. The protocol is strictly cross-antigen: train on two antigens, validate on a third, and test on the fourth, rotating so that each serves as the held-out system once. We then standardized two metrics: (1) pairwise accuracy and (2) retrieval accuracy and precision at top K, which measure how many of a model's K recommendations are experimentally confirmed strong binders.

MochiBind achieved higher pairwise accuracy than every structure-based baseline on all four held-out antigens, outperforming the closest competitor by almost 10% on average. In terms of ranking performance, MochiBind also achieved the highest retrieval accuracy on all four antigens and the highest retrieval precision (lowest false-positive rate) on three out of four. It also scored 200,000 antibody pairs in roughly 13 seconds on a CPU, a more than 100-fold inference speedup over competing methods that should enable the screening of very large design libraries.

BioDiscovery-Figure01.png
MochiBind architecture and the pairwise-to-ranking evaluation framework. (A) We benchmark on a large pool of antibodies per antigen, varying in mutations and affinity. (B) MochiBind embeds an antibody pair and their shared antigen with a protein language model, projects them into a common latent space, and predicts relative binding strength from the difference. (C) Pairwise winners are aggregated into a global ranking. (D) The ranking recommends top candidates, scored by retrieval accuracy and precision at top K.

Learning to predict antibody properties in context

Proteins that bind tightly to their targets but clump together or degrade in the bloodstream or provoke an immune response are not effective or safe as drugs. Most attempts to predict such properties from biological data encounter the same problem: batch effects, or systematic differences in the way different labs handle samples or conduct experiments that lead to predictable deviations in measurement — deviations known as batch offsets. A model fine-tuned on one lab's data quietly inherits its offsets.

In "Context-aware multi-property antibody predictor: A novel framework integrating text and protein language models", in npj Systems Biology and Applications, we address batch effects during inference. Our model — the context-aware multiproperty antibody predictor, or CA-MAP — takes a prompt containing a variable number of example antibodies with their measured properties, followed by a query antibody and the name of the property to predict. When the examples come from the same lab as the query, their measured properties capture the batch offset. The model’s input — its context — thus includes the information it needs to adjust for batch effects without retraining.

Getting a model to use that context, however, is not straightforward. A model trained on data from a single source can learn to ignore the examples — whose measurements are systematically skewed, after all — and rely on the query sequence alone. Our training strategy, AB-context-aware, prevents this by applying a hidden random transformation to both the context properties and the expected answer, resampled for every prompt. Under this scheme, the transformation can be recovered only from the context, so the model must use it.

We measured the effect on a fine-tuned domain-specific multimodal LLM, TxGemma, predicting hydrophobicity. Without batch effects, standard fine-tuning and AB-context-aware training perform comparably, a correlation with ground truth of 0.99 (according to Spearman’s rank correlation coefficient, where 1 is perfect correlation). With a simulated additive batch effect in the 0–0.3 range, standard fine tuning falls to a 0.58 correlation, while the context-aware model remains at 0.99.

CA-MAP has a relatively small multimodal architecture combining text and proteins. Sequences (encoded with ESM-2), property names (encoded with sentence embeddings), and numerical values each have dedicated encoders and projectors, and a state space model based on the sequence-modeling architecture MAMBA composes them. Trained on a synthetic dataset of 876,898 antibody-heavy chains covering six developability properties, CA-MAP achieves a Spearman correlation (denoted ρ) greater than 0.8 on several of them and outperforms the fine-tuned TxGemma baseline across all four properties tested jointly.

The architecture is also considerably cheaper to train and run, with roughly 182,000 trainable parameters to TxGemma’s 40 million, and it’s about 200 times as fast per prompt at inference.

Because properties are specified as text, CA-MAP can also be queried for properties absent from its training data. In one set of experiments, we trained CA-MAP on only four of the dataset’s six developability properties and tested it on the other two (positive-charge heterogeneity, or PosCh, and immunogenicity). When we used only the two target properties as context, immunogenicity prediction reached ρ = 0.25; with all six correlated properties in the context, ρ = 0.73. PosCh improved from ρ = 0.08 to ρ = 0.73 under the same comparison. These gains indicate that the model is drawing on correlations between developability properties, which suggests that expensive assays could be estimated in part from cheaper ones.

BioDiscovery-Figure02.png
CA-MAP architecture, tokenization scheme, and unseen-property results.

Designing antibodies with AI, validating them in the lab

In our third paper, "Agent-guided de novo design of nanobody binders against a novel cancer target", which was presented as a Spotlight at the ICML 2026 Workshop on Generative and Agentic AI for Biology and received the Best Paper Runner-Up Award, we bring predictive and generative antibody models together to design therapeutic nanobodies from scratch in a real drug discovery project.

The target antigen for the design project — or “campaign”, as it’s known in the industry — was chosen to reflect real clinical need: a cell surface target for desmoplastic small round-cell tumors, a rare and aggressive pediatric cancer. Our collaborators at the Dr. Nai-Kong V. Cheung’s Lab at Memorial Sloan Kettering Cancer Center in New York identified it by sequencing patient tumor specimens for proteins that (1) sit on the tumor cell surface, (2) are driven by a specific genetic error, and (3) are largely absent from healthy tissue. The target has no experimental structure and no public antibody information, so there was no template to graft, no prior campaign to affinity-mature from, and no possibility that the design models encountered this antigen during training.

One of the key decisions at the outset of a de novo design campaign is which specific regions on the antigen surface, known as epitopes or hotspots, to target. We designed a hotspot recommendation agent that orchestrates seven bioinformatics tools, which do things like determine solvent-accessible surface area, secondary structure, hydrophobicity, and sequence uniqueness against user-specified negative targets; match epitopes against 500,000 entries in NIAID’s Immune Epitope Database; and annotate domains according to the categories in the protein families (Pfam) database. Our model synthesizes these tools’ outputs into hotspot recommendations with an explicit biophysical rationale for each.

Grounding the recommendations in deterministic tool outputs focuses the search on evidence-supported regions rather than relying on the model's parametric knowledge of protein biology. Evaluated on antibody-antigen complexes from the SAbDab benchmark, the agent recovered at least one true epitope residue within its top five proposed regions about 80% of the time on a diverse holdout set. For the target antigen in our design campaign, it proposed eight hotspot regions.

We then used three generative models with different design principles — RFantibody (diffusion over protein backbones), IgGM (joint sequence-structure diffusion), and mBER (backpropagation through a structure prediction model) — to generate antibody designs that target those hotspots. Each model produced 96,000 designs, and each design was scored on properties like folding confidence (how likely the antibody is to fold into the shape necessary to bind to the target), complex quality (how likely the antibody is to form the correct binding interface with the target), and sequence liabilities (how likely the antibody sequence is to cause development or manufacturing problems), and MochiBind's sequence-based affinity estimate. Our candidate selection agent applied multi-objective Pareto filtering to ensure the retention of designs excelling on different metric combinations, and it prioritized 100,000 candidates for experimental screening.

Each candidate was synthesized and displayed on the surface of a yeast cell to be screened for whether it stuck to the target, and the designs that stuck most strongly were carried forward through two rounds of sorting and filtering. None of the 116 candidates that survived these rounds bound to an unrelated control protein, indicating that they bind specifically to the intended target, rather than being generally sticky. All 116 were then individually measured to determine how tightly they bind to the target antigen, and 46 were identified as strong binders.

These 46 binders, along with the binder and nonbinder labels from the full screen, become training data for the next design cycle: a lab-in-the-loop workflow where each round of experiments sharpens the models that propose the following round. Amazon is uniquely well positioned to run that loop , with the scientific expertise to build foundational ML for biology, the computational capacity to design and score hundreds of thousands of candidates, and a path to deliver these methods, including those like MochiBind and CA-MAP that aren’t available today, to customers through Amazon Bio Discovery, an AI-powered application that connects these biological AI models with integrated lab services so scientists can move from design to experimental validation in a single workflow.

BioDiscovery-Figure03.png
Integrated computational and experimental pipeline for de novo nanobody design.

Research areas

Related content

US, VA, Arlington
The Sponsored Products and Brands team at Amazon Ads is re-imagining the advertising landscape through SOTA generative AI technologies, revolutionizing how millions of customers discover products and engage with brands across Amazon.com and beyond. We are at the forefront of re-inventing advertising experiences, bridging human creativity with artificial intelligence to transform every aspect of the advertising lifecycle from ad creation and optimization to performance analysis and customer insights. We are a passionate group of innovators dedicated to developing responsible and intelligent AI technologies that balance the needs of advertisers, enhance the shopping experience, and strengthen the marketplace. If you're energized by solving complex challenges and pushing the boundaries of what's possible with AI, join us in shaping the future of advertising. The Marketplace Intelligence (MI) team is looking for an Applied Science Manager to lead a team of scientists and engineers in building production ML and bandit solutions to customize the search experience. We determine which ads to show in Amazon search, where to place them, how many ads to place, and to which customers. This helps shoppers discover new products while helping advertisers put their products in front of the right customers, aligning shoppers’, advertisers’, and Amazon’s interests. To do this, we apply a broad range of machine learning, causal inference, and optimization techniques to continuously explore, learn, and optimize the allocation and ranking of ads on the search page. We are an interdisciplinary team with a focus on customer obsession and inventing and simplifying. Our primary focus is on improving the SP experience in search by gaining a deep understanding of shopper pain points and developing new innovative solutions to address them. You’ll lead the MI Interleaving team. The Interleaving team’s mission is to personalize and contextualize SP ad allocation on the search page. We do this by modeling shopper responses to the number, placement, and quality of ads. We use online experimentation, simulation, causal modeling, and online feedback to estimate the cost of displacing organic and sponsored content. Then, we incorporate those estimates into ad allocation to deliver an efficient and customized shopping experience for shoppers and improved discoverability and sales for advertisers. You’ll own the experimentation systems, models, and online model serving infrastructure to support these solutions. This is a unique opportunity for someone who wants to have broad business impact, a direct impact on customers and the search experience, build scaled real-time LLM and ML solutions, and lead a cross-functional team. If you are interested in machine learning, bandit learning, building production systems, and leading a team to build these solutions, this role is for you. We’re looking for a leader who can help lay out the vision for the team and grow with it. Key job responsibilities * Lead a team of scientists and engineers in building scalable machine learning solutions. * Develop a vision for contextualizing and personalizing SP ads in Amazon search. * Create, develop, and drive a data-driven product strategy to define the right quantitative measures of shopper impact, using this to evaluate decisions and opportunities. * Tackle and solve challenging science and business problems that balance the interests of advertisers, shoppers, and Amazon. * Own a portfolio of pragmatic long-term investments that drive long-term growth of the ads and retail businesses. * Develop real-time LLM and ML algorithms to allocate billions of ads per day in advertising auctions. * Develop efficient algorithms for multi-objective optimization and AI control methods to find operating points for the ad marketplace then evolve them * Develop scientists and ML engineers around machine learning, economics, and optimization for Advertising.
US, CA, Culver City
MULTIPLE POSITIONS AVAILABLE Employer: AMAZON.COM SERVICES LLC Offered Position: Data Scientist II Job Location: Culver City, California Job Number: AMZ10564056 Position Responsibilities: Design and implement scalable and reliable approaches to support or automate decision making throughout the business. Apply a range of data science techniques and tools combined with subject matter expertise to solve difficult business problems and cases in which the solution approach is unclear. Acquire data by building the necessary SQL / ETL queries. Import processes through various company specific interfaces for accessing Oracle, RedShift, and Spark storage systems. Build relationships with stakeholders and counterparts. Analyze data for trends and input validity by inspecting univariate distributions, exploring bivariate relationships, constructing appropriate transformations, and tracking down the source and meaning of anomalies. Build models using statistical modeling, mathematical modeling, econometric modeling, network modeling, social network modeling, natural language processing, machine learning algorithms, genetic algorithms, and neural networks. Validate models against alternative approaches, expected and observed outcome, and other business defined key performance indicators. Implement models that comply with evaluations of the computational demands, accuracy, and reliability of the relevant ETL processes at various stages of production. 40 hours / week, 8:00am-5:00pm, Salary Range: $158,681/year to $184,000/year. Amazon is a total compensation company. Dependent on the position offered, equity, sign-on payments, and other forms of compensation may be provided as part of a total compensation package, in addition to a full range of medical, financial, and/or other benefits. For more information, visit: https://www.aboutamazon.com/workplace/employee-benefits. Amazon.com is an Equal Opportunity-Affirmative Action Employer – Minority / Female / Disability / Veteran / Gender Identity / Sexual Orientation.#0000
CH, Zurich
RIVR, an Amazon company, is building Physical AI by deploying autonomous robots for real-world doorstep delivery. Operating daily in diverse urban environments, RIVR's robots continuously learn from and navigate the millions of scenarios encountered during deliveries. By owning the full stack from software. Our fleet of delivery robots operates globally today, generating vast amounts of robotic real-world data. By utilizing state-of-the-art Vision-Language-Action (VLA) models, large-scale generalist models (like Transformers), generative AI, and similar methods, we can leverage this pool of data to significantly enhance its autonomy, navigation, and manipulation skills. In this role, you will develop multi-modal models that enable robots to autonomously generate actions from demonstrations, real-time sensor data, and natural language commands. We are seeking an expert in VLA models, imitation learning, and generative AI techniques with a deep knowledge of supervised, and self-supervised learning algorithms. If you are passionate about pushing the boundaries of AI we invite you to join us in shaping the future of intelligent robotics. Key job responsibilities Develop and implement Vision-Language-Action (VLA) models, generalist robot transformers, and imitation learning algorithms (e.g., diffusion policies) to enable robots to autonomously execute complex tasks. Design, test, and refine your algorithms to meet the demands of complex real-world autonomy and navigation tasks, with a focus on spatial reasoning and generalization. Streamline the data collection and training workflow to efficiently expand model capabilities with new tasks and data sources. Collaborate with the reinforcement learning team to innovate methods that leverage both simulated and real-world data. Optimize and distill networks for real-time deployment on the edge (e.g. Nvidia Jetson Thor). Build, lead and mentor an exceptional team of software engineers. Provide expert guidance to product managers and executives for strategic decision-making. Create and maintain documentation, guidelines, and best practices to streamline knowledge sharing.
ES, M, Madrid
At Amazon, we are committed to being the Earth’s most customer-centric company. The European International Technology group (EU INTech) owns the enhancement and delivery of Amazon’s engineering to all the varied customers and cultures of the world. We do this through a combination of partnerships with other Amazon technical teams and our own innovative new projects. You will be joining the Tools and Machine Learning (Tamale) team. As part of EU INTech, we tackle complex, high-impact problems at the intersection of e-commerce, AI, and customer experience; going well beyond traditional ML pipelines. Our work spans multiple frontiers: we build GenAI-powered data solutions, enabling scalable and cost-efficient training data generation at Amazon scale. We design and deploy agentic systems for detecting and fixing quality issues across Amazon's retail surfaces. And we develop ranking and recommendation models for Haul shopping experiences. We are looking for a passionate, talented, and inventive Scientist with a strong machine learning background to help build industry-leading machine learning solutions. We strongly value your hard work and obsession to solve complex problems on behalf of Amazon customers. Key job responsibilities We look for applied scientists who possess a wide variety of skills. As the successful applicant for this role, you will with work closely with your business partners to identify opportunities for innovation. You will apply machine learning solutions to automate manual processes, to scale existing systems and to improve catalog data quality, to name just a few. You will work with business leaders, scientists, and product managers to translate business and functional requirements into concrete deliverables, including the design, development, testing, and deployment of highly scalable distributed services. You will be part of team of scientists and engineers working on solving data quality issues at scale. You will be able to influence the scientific roadmap of the team, setting the standards for scientific excellence. You will be working with state-of-the-art models, including image to text, Multimodal LLMs and GenAI. Your work will improve the experience of millions of daily customers using Amazon in Europe and in other regions. You will have the chance to have great customer impact and continue growing in one of the most innovative companies in the world. You will learn a huge amount - and have a lot of fun - in the process!
CL, Virtual
The Central Science Team within Amazon’s People Experience and Technology org (PXTCS) uses economics, behavioral science, statistics, and machine learning to proactively identify mechanisms and process improvements which simultaneously improve Amazon and the lives, well-being, and the value of work to Amazonians. We are an interdisciplinary team, which combines the talents of science and engineering to develop and deliver solutions that measurably achieve this goal. We are looking for a Economist II who is able to provide structure around complex business problems, hone those complex problems into specific, scientific questions, and test those questions to generate insights. The ideal candidate will work with various science, engineering, operations, and analytics teams to estimate models and algorithms on large scale data, design pilots and measure their impact, and transform successful prototypes into improved policies and programs at scale. They will lead teams of researchers to produce robust, objective research results and insights which can be communicated to a broad audience inside and outside of Amazon. The ideal candidate has a PhD in Economics and deep expertise in causal inference and applied econometrics. Experience with large-scale data, proficiency in statistical programming (Python), and familiarity with machine learning methods are a plus. To be successful in this role, you should be comfortable operating with ambiguity, able to independently scope and prioritize research agendas, skilled at influencing decisions through rigorous analysis, and comfortable with using AI tools.
US, MA, Boston
Our team is involved with pre-silicon design verification for custom IP. A critical requirement of the verification flow is the requirement of legal and realistic stimulus of a custom Machine Learning Accelerator Chip. Content creation is built using formal methods that model legal behavior of the design and then solving the problem to create the specific assembly tests. The entire frame work for creating these custom tests is developed using a SMT solver and custom software code to guide the solution space into templated scenarios. This highly visible and innovative role requires the design of this solving framework and collaborating with design verification engineers, hardware architects and designers to ensure that interesting content can be created for the projects needs. Key job responsibilities Develop an understanding for a custom machine learning instruction set architecture. Model correctness of instruction streams using first order logic. Create custom API's to allow control over scheduling and randomness. Deploy algorithms to ensure concurrent code is safely constructed. Create coverage metrics to ensure solution space coverage. Use novel methods like machine learning to automate content creation. About the team Utility Computing (UC) AWS Utility Computing (UC) provides product innovations — from foundational services such as Amazon’s Simple Storage Service (S3) and Amazon Elastic Compute Cloud (EC2), to consistently released new product innovations that continue to set AWS’s services and features apart in the industry. As a member of the UC organization, you’ll support the development and management of Compute, Database, Storage, Internet of Things (Iot), Platform, and Productivity Apps services in AWS, including support for customers who require specialized security solutions for customers who require specialized security solutions for their cloud services. Annapurna Labs (our organization within AWS UC) designs silicon and software that accelerates innovation. Customers choose us to create cloud solutions that solve challenges that were unimaginable a short time ago—even yesterday. Our custom chips, accelerators, and software stacks enable us to take on technical challenges that have never been seen before, and deliver results that help our customers change the world.
IN, KA, Bengaluru
Have you ever wondered how that Amazon box with the smile arrives so quickly, where it came from, and how much it cost Amazon to deliver? The WW Amazon Logistics, Business Analytics team manages 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 seeking an enthusiastic, customer-obsessed Research Science Manager(RSM) with strong analytical skills to join our team. This role is crucial in optimizing Amazon's vast delivery network and will have significant impact on the customer experience, particularly in the final phase of delivery. As a Research Science Manager, you will: 1. Address business challenges through building compelling cases and using data to influence change across the organization 2. Develop input and assumptions based on preexisting models to estimate costs and savings opportunities associated with varying levels of network growth and operations 3. Create metrics to measure business performance, identify root causes and trends, and prescribe action plans 4. Manage multiple high-impact projects simultaneously 5. Work with technology teams and product managers to develop new tools and systems supporting business growth 6. Communicate with and support various internal stakeholders and external audiences 7. Implement scheduling solutions, improve metrics, and develop scalable processes and tools Looking for minds with: Extensive experience in operations research and data-driven decision making Strong analytical and problem-solving skills Robust program management and research science skills Ability to work with a team and make independent decisions in ambiguous environments Customer-obsessed mindset with a focus on improving the Amazon delivery experience This role offers the autonomy to think strategically and make data-driven decisions from day one. Join us in shaping the future of e-commerce delivery and addressing the core challenges in our world-class operations space! Key job responsibilities Key job responsibilities 1. Advanced Modeling and Algorithm Development: Design and implement sophisticated machine learning models for logistics optimization Develop complex time series forecasting algorithms for demand prediction and resource allocation 2. AI and Machine Learning Integration: Architect and deploy AI-powered systems to enhance decision-making in logistics operations Implement deep learning techniques for image recognition in package sorting and handling Develop reinforcement learning algorithms for adaptive scheduling and resource management 3. Big Data Analytics and Processing: Design and implement distributed computing solutions for processing massive logistics datasets Utilize cloud computing platforms (e.g., AWS) for scalable data processing and analysis 4. AI-Driven Workflow Optimization: Design and implement AI agents for autonomous decision-making in logistics processes Create machine learning models for customer behavior analysis and personalized delivery options 5. Software Development and System Architecture: Write efficient, scalable code in languages such as Python, Java, or C++ Develop and maintain complex software systems for logistics optimization Stay at the forefront of AI and ML research Publish research findings in top-tier conferences and journals About the team We are Amazon's Last Mile Science and Analytics team, dedicated to improving e-commerce delivery. We work to optimize our vast network, forecast demand using machine learning, and enhance route efficiency. Our efforts focus on developing innovative delivery methods, applying AI to solve complex problems, and conducting geospatial analysis. We create simulations to refine processes and plan capacity effectively. Operating globally, we strive to develop adaptable solutions for diverse markets. We aim to advance logistics science, continually improving speed, efficiency, and customer satisfaction, in support of Amazon's mission to be Earth's most customer-centric company.
IN, KA, Bengaluru
Amazon Devices is an inventive research and development company that designs and engineer high-profile devices like the Kindle family of products, Fire Tablets, Fire TV, Health Wellness, Amazon Echo & Astro products. This is an exciting opportunity to join Amazon in developing state-of-the-art techniques that bring Gen AI on edge for our consumer products. We are looking for exceptional scientists to join our Applied Science team and help develop the next generation of edge models, and optimize them while doing co-designed with custom ML HW based on a revolutionary architecture. Work hard. Have Fun. Make History. Key job responsibilities - Quantize, prune, distill, finetune Gen AI models to optimize for edge platforms - Fundamentally understand Amazon’s underlying Neural Edge Engine to invent optimization techniques - Analyze deep learning workloads and provide guidance to map them to Amazon’s Neural Edge Engine - Use first principles of Information Theory, Scientific Computing, Deep Learning Theory, Non Equilibrium Thermodynamics - Train custom Gen AI models that beat SOTA and paves path for developing production models - Collaborate closely with compiler engineers, fellow Applied Scientists, Hardware Architects and product teams to build the best ML-centric solutions for our devices - Publish in open source and present on Amazon's behalf at key ML conferences - NeurIPS, ICLR, MLSys.
US, CA, Culver City
MULTIPLE POSITIONS AVAILABLE Employer: AMAZON.COM SERVICES LLC Offered Position: Applied Scientist III Job Location: Culver City, California Job Number: AMZ10564141 Position Responsibilities: Participate in the design, development, evaluation, deployment and updating of data-driven models and analytical solutions for machine learning (ML) and/or natural language (NL) applications. Develop and/or apply statistical modeling techniques (e.g. Bayesian models and deep neural networks), optimization methods, and other ML techniques to different applications in business and engineering. Routinely build and deploy ML models on available data, and run and analyze experiments in a production environment. Identify new opportunities for research in order to meet business goals. Research and implement novel ML and statistical approaches to add value to the business. Mentor junior engineers and scientists. 40 hours / week, 8:00am-5:00pm, Salary Range: $167,100/year to $226,100/year. Amazon is a total compensation company. Dependent on the position offered, equity, sign-on payments, and other forms of compensation may be provided as part of a total compensation package, in addition to a full range of medical, financial, and/or other benefits. For more information, visit: https://www.aboutamazon.com/workplace/employee-benefits. Amazon.com is an Equal Opportunity-Affirmative Action Employer – Minority / Female / Disability / Veteran / Gender Identity / Sexual Orientation.#0000
IN, KA, Bengaluru
We are looking for a talented Applied Scientist to join our team. In this role, you will design, develop, and deploy machine learning and computer vision models that solve real-world problems at scale in the Amazon grocery domain. You will work closely with engineering, product, and business teams to turn complex technical challenges into production-ready solutions, and own the model development lifecycle from experimentation through deployment. You will bring scientific rigor to every stage — from data analysis and model design to evaluation and iteration. This is a high-impact role where your models will directly improve the shopping experience for millions of customers in Amazon grocery stores. Key job responsibilities Design, train, and evaluate computer vision and machine learning models for complex grocery-domain problems including product identification, shelf perception, and in-store scene understanding — iterating rapidly from prototype to production-quality solutions Conduct rigorous exploratory data analysis to characterize domain-specific challenges (image variability, catalog gaps, label noise) and translate findings into actionable modeling decisions Own the model development lifecycle from experimentation through deployment — collaborating with software and ML engineers to ensure models meet latency, throughput, and reliability requirements at production scale Design and execute offline and online evaluation frameworks — defining metrics that capture both model performance and downstream business impact, and diagnosing failure modes to prioritize improvements Build and improve data pipelines and annotation workflows that feed model training, including active learning strategies to maximize label efficiency Communicate technical results, trade-offs, and recommendations clearly to engineering, product, and business stakeholders — connecting model behavior to customer experience outcomes Stay current with state-of-the-art research in computer vision, multimodal learning, and representation learning — evaluating and adapting promising techniques to team-specific problems Contribute to a culture of scientific rigor through reproducible experimentation, thorough documentation, peer code and design reviews, and raising the quality bar for the team A day in the life As an Applied Scientist on the GRAISE team, you'll spend your days analyzing model performance from overnight experiments, collaborating with engineers to deploy computer vision models to production, and prototyping new approaches using multimodal learning with store video and sensor data. You'll present findings to product and business stakeholders, translating technical results into actionable recommendations. Throughout the day, you'll balance rigorous scientific thinking with practical engineering constraints, knowing your work directly improves the shopping experience for millions of customers in Amazon grocery stores. About the team The GRAISE team (Grocery, Retail & In-Store Experience) within World Wide Grocery Store Tech (WWGST) builds foundational AI and machine learning systems that power Amazon's in-store grocery technologies. We develop domain-specific models that solve uniquely complex challenges in grocery — from smart shopping carts and inventory intelligence to personalization and store operations. Our mission is to create technology which makes grocery shopping more convenient, economical, personalized, and enjoyable for customers while empowering retailers with operational efficiency