Inderjit Dhillon and Michael I. Jordan
The collaboration between Berkeley BAIR and Amazon brings together Inderjit Dhillon (l), and Michael I. Jordan. Dhillon is a professor of computer science and mathematics at the University of Texas Austin, and an Amazon Fellow; Jordan is a professor of machine learning, statistics, and AI at UC Berkeley, and an Amazon Scholar.
Credit: University of Texas at Austin, and Flavia Loreto

Collaboration between Amazon and UC Berkeley advances AI and machine learning

For two years now, Amazon has been collaborating with the University of California Berkeley Artificial Intelligence Research (BAIR) Lab, giving students the opportunity to work on challenging, real-world problems.

In a keynote talk at the 2019 re:MARS Conference, Jeff Wilke, CEO of Amazon’s worldwide consumer business, talked about how the company was built by computer scientists and incredible coders, but increasingly “we’re being propelled forward by the work of pioneering AI scientists” – scientists who are shaping customer experiences in shopping, delivery, voice (Alexa), and many other areas.

Within the world of machine learning and AI, one of the premiere institutions of study and research is the University of California, Berkeley. Its impact on computer science has been profound, with groundbreaking work in theoretical as well as practical aspects of technology, including control theory, pattern recognition, neural networks, computer vision, and robotics. The school’s BAIR (Berkeley Artificial Intelligence Research) Lab brings together UC Berkeley researchers across the areas of machine learning, natural language processing, AI, and other topics.

For two years now, Amazon has been collaborating with UC Berkeley and BAIR on a wide range of research projects. This collaboration represents a win-win opportunity for both organizations, giving Amazon a chance to work with to some of the brightest computer scientists in academia, while providing UC Berkeley students an opportunity to tackle challenging, real-world problems.

Berkeley BAIR and Amazon logos
For two years now, Amazon has been collaborating with UC Berkeley and BAIR on a wide range of research projects.

Currently eight specific projects – covering machine learning, econometrics, Bayesian statistics, and more – are part of the collaboration.

“Machine learning and AI have become core to everything Amazon does,” says Inderjit Dhillon, an Amazon Fellow in Search and AI who is working with UC Berkeley researchers. “In these areas, UC Berkeley is arguably the world’s leading university. It just seemed natural to us to start a conversation with them and see what we could work on together.”

The collaboration brings together some heavyweight computer science intellectual horsepower.

In addition to his role as Amazon Fellow, Dhillon is the Gottesman Family Centennial Professor of Computer Science and Mathematics at the University of Texas at Austin, where he is also the director of the ICES Center for Big Data Analytics. Michael Jordan, an Amazon Scholar, runs the Berkeley side of the collaboration. He is a professor of machine learning, statistics, and AI at UC Berkeley, and in 2016 was recognized as the world’s most influential computer scientist by Science magazine. Their collective intellectual standing is noteworthy; both Dhillon and Jordan are among the top five computer science professors in terms of publications over the last decade at leading machine learning conferences, such as the Conference on Neural Information Processing Systems.

Jordan says that Amazon presents one of the more interesting real-world users of machine learning and AI in the world.

“They were one of the first companies to use machine learning for real problems like supply-chain management and recommendation engines,” he says. “Amazon is uniquely suited to take cutting-edge academic research and put it to work in a meaningful way.”

For Berkeley students, the collaboration with Amazon gives them a chance to use their machine learning and AI skills to help solve some of the big challenges Amazon is seeking to address.

“We like to work on things that are really used in industry,” says Jordan. “The work with Amazon is relevant by definition, which makes the projects students work on so much more pertinent and worthwhile.”

“Students love working with Amazon,” he adds. “They think it is an absolute win-win. They get to see behind the website, getting a glimpse of the vast array of underlying problems in search, pricing, logistics, services, and systems – they see the whole Amazon system end-to-end.”

Examples of some of the projects students are working on include one related to visual search that utilizes a handheld camera phone to identify and extract semantic objects from a scene; another focused on time series forecasting with limited data that’s applicable to retail demand forecasting, and other areas of Amazon’s business; and still another focused on the use of probabilistic models at scale. All the work done by students is performed under the university’s open-research model, meaning its non-exclusive, and open publication and open-source code release is expected.

Over the next few decades we’ll see machine-learning systems continue to grow in scope and scale, changing the economy, and a new academic field will emerge to support that growth. Its emergence will reflect a deep partnership between industry and academia, with benefits for everyone.
Michael I. Jordan, UCal Berkeley professor and Amazon Scholar

Pierre Brunelle, the Amazon technical product manager who scoped and launched the eight projects the partnership now is working on, says that such collaborations help companies such as Amazon find new perspectives on challenging problems.

“Advances in applied science have often come from a strong partnership between the research and industry worlds,” he says. “The future of machine learning and AI calls for higher systems requirements, in terms of scalability, performance and adaptability. All our projects work towards these goals.”

Brian Burdick, Amazon’s director of Core AI, manages the partnership for the company, and agrees the collaboration is mutually beneficial.

Brian Burdick
Brian Burdick, Amazon director of Core AI

“We have had ‘Amazon meets Berkeley Days’ when we proposed problems to the students and brainstormed projects with them – a dialogue that has been really helpful,” he says. “Working with Amazon gives students a chance to see the entire lifecycle of a project and build in each step. That helps inform their more theoretical or open-ended research.”

Amazon also benefits by opening a more direct recruiting pipeline to some of the world’s best computer-science students. Seven students have joined Amazon in some capacity since the partnership began, several as full-time employees.

What’s next for this partnership, and others like it?

Burdick says he hopes to see collaborations such as the one between Amazon and UC Berkeley expand to other institutions in the years to come, combining the enthusiasm and academic smarts of students with the real-world nature of research at Amazon.

“I’d like to see a lifecycle of working with grad students and research faculty from collaborative on-campus research, to internships with us, then see that real-world knowledge flow back into academia to inspire the next generation of science,” he says.

Adds Jordan: “I think that it’s helpful to think back to the 1920s and 1930s, when there was no such field as chemical engineering. People and that era began to imagine that it was possible to build large-scale chemical plants, doing chemistry at a new, economically-relevant scale. Accordingly, an academic field developed to support such efforts. That is exactly where we are currently with machine learning. Over the next few decades we’ll see machine-learning systems continue to grow in scope and scale, changing the economy, and a new academic field will emerge to support that growth. Its emergence will reflect a deep partnership between industry and academia, with benefits for everyone.”

Research areas

Related content

GB, MLN, Edinburgh
We’re looking for a Machine Learning Scientist in the Personalization team for our Edinburgh office experienced in generative AI and large models. You will be responsible for developing and disseminating customer-facing personalized recommendation models. This is a hands-on role with global impact working with a team of world-class engineers and scientists across the Edinburgh offices and wider organization. You will lead the design of machine learning models that scale to very large quantities of data, and serve high-scale low-latency recommendations to all customers worldwide. You will embody scientific rigor, designing and executing experiments to demonstrate the technical efficacy and business value of your methods. You will work alongside a science team to delight customers by aiding in recommendations relevancy, and raise the profile of Amazon as a global leader in machine learning and personalization. Successful candidates will have strong technical ability, focus on customers by applying a customer-first approach, excellent teamwork and communication skills, and a motivation to achieve results in a fast-paced environment. Our position offers exceptional opportunities for every candidate to grow their technical and non-technical skills. If you are selected, you have the opportunity to make a difference to our business by designing and building state of the art machine learning systems on big data, leveraging Amazon’s vast computing resources (AWS), working on exciting and challenging projects, and delivering meaningful results to customers world-wide. Key job responsibilities Develop machine learning algorithms for high-scale recommendations problems. Rapidly design, prototype and test many possible hypotheses in a high-ambiguity environment, making use of both quantitative analysis and business judgement. Collaborate with software engineers to integrate successful experimental results into large-scale, highly complex Amazon production systems capable of handling 100,000s of transactions per second at low latency. Report results in a manner which is both statistically rigorous and compellingly relevant, exemplifying good scientific practice in a business environment.
IN, TS, Hyderabad
Welcome to the Worldwide Returns & ReCommerce team (WWR&R) at Amazon.com. WWR&R is an agile, innovative organization dedicated to ‘making zero happen’ to benefit our customers, our company, and the environment. Our goal is to achieve the three zeroes: zero cost of returns, zero waste, and zero defects. We do this by developing products and driving truly innovative operational excellence to help customers keep what they buy, recover returned and damaged product value, keep thousands of tons of waste from landfills, and create the best customer returns experience in the world. We have an eye to the future – we create long-term value at Amazon by focusing not just on the bottom line, but on the planet. We are building the most sustainable re-use channel we can by driving multiple aspects of the Circular Economy for Amazon – Returns & ReCommerce. Amazon WWR&R is comprised of business, product, operational, program, software engineering and data teams that manage the life of a returned or damaged product from a customer to the warehouse and on to its next best use. Our work is broad and deep: we train machine learning models to automate routing and find signals to optimize re-use; we invent new channels to give products a second life; we develop highly respected product support to help customers love what they buy; we pilot smarter product evaluations; we work from the customer backward to find ways to make the return experience remarkably delightful and easy; and we do it all while scrutinizing our business with laser focus. You will help create everything from customer-facing and vendor-facing websites to the internal software and tools behind the reverse-logistics process. You can develop scalable, high-availability solutions to solve complex and broad business problems. We are a group that has fun at work while driving incredible customer, business, and environmental impact. We are backed by a strong leadership group dedicated to operational excellence that empowers a reasonable work-life balance. As an established, experienced team, we offer the scope and support needed for substantial career growth. Amazon is earth’s most customer-centric company and through WWR&R, the earth is our customer too. Come join us and innovate with the Amazon Worldwide Returns & ReCommerce team!
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 add-on subscriptions such as Apple TV+, Max, 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 technologist, 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! In Prime Video READI, our mission is to automate infrastructure scaling and operational readiness. We are growing a team specialized in time series modeling, forecasting, and release safety. This team will invent and develop algorithms for forecasting multi-dimensional related time series. The team will develop forecasts on key business dimensions with optimization recommendations related to performance and efficiency opportunities across our global software environment. As a founding member of the core team, you will apply your deep coding, modeling and statistical knowledge to concrete problems that have broad cross-organizational, global, and technology impact. Your work will focus on retrieving, cleansing and preparing large scale datasets, training and evaluating models and deploying them to production where we continuously monitor and evaluate. You will work on large engineering efforts that solve significantly complex problems facing global customers. You will be trusted to operate with complete independence and are often assigned to focus on areas where the business and/or architectural strategy has not yet been defined. You must be equally comfortable digging in to business requirements as you are drilling into design with development teams and developing production ready learning models. You consistently bring strong, data-driven business and technical judgment to decisions. You will work with internal and external stakeholders, cross-functional partners, and end-users around the world at all levels. Our team makes a big impact because nothing is more important to us than delivering for our customers, continually earning their trust, and thinking long term. You are empowered to bring new technologies to your solutions. If you crave a sense of ownership, this is the place to be.
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 add-on subscriptions such as Apple TV+, Max, 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 As an Applied Scientist in the Content Understanding Team, you will lead the end-to-end research and deployment of video and multi-modal models applied to a variety of downstream applications. More specifically, you will: - Work backwards from customer problems to research and design scientific approaches for solving them - Work closely with other scientists, engineers and product managers to expand the depth of our product insights with data, create a variety of experiments to determine the high impact projects to include in planning roadmaps - Stay up-to-date with advancements and the latest modeling techniques in the field - Publish your research findings in top conferences and journals About the team Our Prime Video Content Understanding team builds holistic media representations (e.g. descriptions of scenes, semantic embeddings) and apply them to new customer experiences supply chain problems. Our technology spans the entire Prime Video catalogue globally, and we enable instant recaps, skip intro timing, ad placement, search, and content moderation.
US, MA, Boston
The Artificial General Intelligence (AGI) team is looking for a passionate, talented, and inventive Applied Scientist with a strong deep learning background, to build industry-leading technology with Large Language Models (LLMs) and multi-modal systems. You will support projects that work on technologies including multi-modal model alignment, moderation systems and evaluation. Key job responsibilities As an Applied Scientist with the AGI team, you will support the development of novel algorithms and modeling techniques, to advance the state of the art with LLMs. Your work will directly impact our customers in the form of products and services that make use of speech and language technology. You will leverage Amazon’s heterogeneous data sources and large-scale computing resources to accelerate advances in generative artificial intelligence (GenAI). You are also expected to publish in top tier conferences. About the team The AGI team has a mission to push the envelope in LLMs and multimodal systems. Specifically, we focus on model alignment with an aim to maintain safety while not denting utility, in order to provide the best-possible experience for our customers.
US, WA, Seattle
Do you want to re-invent how millions of people consume video content on their TVs, Tablets and Alexa? We are building a free to watch streaming service called Fire TV Channels (https://techcrunch.com/2023/08/21/amazon-launches-fire-tv-channels-app-400-fast-channels/). Our goal is to provide customers with a delightful and personalized experience for consuming content across News, Sports, Cooking, Gaming, Entertainment, Lifestyle and more. You will work closely with engineering and product stakeholders to realize our ambitious product vision. You will get to work with Generative AI and other state of the art technologies to help build personalization and recommendation solutions from the ground up. You will be in the driver's seat to present customers with content they will love. Using Amazon’s large-scale computing resources, you will ask research questions about customer behavior, build state-of-the-art models to generate recommendations and run these models to enhance the customer experience. You will participate in the Amazon ML community and mentor Applied Scientists and Software Engineers with a strong interest in and knowledge of ML. Your work will directly benefit customers and you will measure the impact using scientific tools.
IN, HR, Gurugram
We're on a journey to build something new a green field project! Come join our team and build new discovery and shopping products that connect customers with their vehicle of choice. We're looking for a talented Senior Applied Scientist to join our team of product managers, designers, and engineers to design, and build innovative automotive-shopping experiences for our customers. This is a great opportunity for an experienced engineer to design and implement the technology for a new Amazon business. We are looking for a Applied Scientist to design, implement and deliver end-to-end solutions. We are seeking passionate, hands-on, experienced and seasoned Senior Applied Scientist who will be deep in code and algorithms; who are technically strong in building scalable computer vision machine learning systems across item understanding, pose estimation, class imbalanced classifiers, identification and segmentation.. You will drive ideas to products using paradigms such as deep learning, semi supervised learning and dynamic learning. As a Senior Applied Scientist, you will also help lead and mentor our team of applied scientists and engineers. You will take on complex customer problems, distill customer requirements, and then deliver solutions that either leverage existing academic and industrial research or utilize your own out-of-the-box but pragmatic thinking. In addition to coming up with novel solutions and prototypes, you will directly contribute to implementation while you lead. A successful candidate has excellent technical depth, scientific vision, project management skills, great communication skills, and a drive to achieve results in a unified team environment. You should enjoy the process of solving real-world problems that, quite frankly, haven’t been solved at scale anywhere before. Along the way, we guarantee you’ll get opportunities to be a bold disruptor, prolific innovator, and a reputed problem solver—someone who truly enables AI and robotics to significantly impact the lives of millions of consumers. Key job responsibilities Architect, design, and implement Machine Learning models for vision systems on robotic platforms Optimize, deploy, and support at scale ML models on the edge. Influence the team's strategy and contribute to long-term vision and roadmap. Work with stakeholders across , science, and operations teams to iterate on design and implementation. Maintain high standards by participating in reviews, designing for fault tolerance and operational excellence, and creating mechanisms for continuous improvement. Prototype and test concepts or features, both through simulation and emulators and with live robotic equipment Work directly with customers and partners to test prototypes and incorporate feedback Mentor other engineer team members. A day in the life - 6+ years of building machine learning models for retail application experience - PhD, or Master's degree and 6+ years of applied research experience - Experience programming in Java, C++, Python or related language - Experience with neural deep learning methods and machine learning - Demonstrated expertise in computer vision and machine learning techniques.
IN, HR, Gurugram
Our customers have immense faith in our ability to deliver packages timely and as expected. A well planned network seamlessly scales to handle millions of package movements a day. It has monitoring mechanisms that detect failures before they even happen (such as predicting network congestion, operations breakdown), and perform proactive corrective actions. When failures do happen, it has inbuilt redundancies to mitigate impact (such as determine other routes or service providers that can handle the extra load), and avoids relying on single points of failure (service provider, node, or arc). Finally, it is cost optimal, so that customers can be passed the benefit from an efficiently set up network. Amazon Shipping is hiring Applied Scientists to help improve our ability to plan and execute package movements. As an Applied Scientist in Amazon Shipping, you will work on multiple challenging machine learning problems spread across a wide spectrum of business problems. You will build ML models to help our transportation cost auditing platforms effectively audit off-manifest (discrepancies between planned and actual shipping cost). You will build models to improve the quality of financial and planning data by accurately predicting ship cost at a package level. Your models will help forecast the packages required to be pick from shipper warehouses to reduce First Mile shipping cost. Using signals from within the transportation network (such as network load, and velocity of movements derived from package scan events) and outside (such as weather signals), you will build models that predict delivery delay for every package. These models will help improve buyer experience by triggering early corrective actions, and generating proactive customer notifications. Your role will require you to demonstrate Think Big and Invent and Simplify, by refining and translating Transportation domain-related business problems into one or more Machine Learning problems. You will use techniques from a wide array of machine learning paradigms, such as supervised, unsupervised, semi-supervised and reinforcement learning. Your model choices will include, but not be limited to, linear/logistic models, tree based models, deep learning models, ensemble models, and Q-learning models. You will use techniques such as LIME and SHAP to make your models interpretable for your customers. You will employ a family of reusable modelling solutions to ensure that your ML solution scales across multiple regions (such as North America, Europe, Asia) and package movement types (such as small parcel movements and truck movements). You will partner with Applied Scientists and Research Scientists from other teams in US and India working on related business domains. Your models are expected to be of production quality, and will be directly used in production services. You will work as part of a diverse data science and engineering team comprising of other Applied Scientists, Software Development Engineers and Business Intelligence Engineers. You will participate in the Amazon ML community by authoring scientific papers and submitting them to Machine Learning conferences. You will mentor Applied Scientists and Software Development Engineers having a strong interest in ML. You will also be called upon to provide ML consultation outside your team for other problem statements. If you are excited by this charter, come join us!
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 add-on subscriptions such as Apple TV+, Max, 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 As an Applied Scientist in the Content Understanding Team, you will lead the end-to-end research and deployment of video and multi-modal models applied to a variety of downstream applications. More specifically, you will: - Work backwards from customer problems to research and design scientific approaches for solving them - Work closely with other scientists, engineers and product managers to expand the depth of our product insights with data, create a variety of experiments to determine the high impact projects to include in planning roadmaps - Stay up-to-date with advancements and the latest modeling techniques in the field - Publish your research findings in top conferences and journals About the team Our Prime Video Content Understanding team builds holistic media representations (e.g. descriptions of scenes, semantic embeddings) and apply them to new customer experiences supply chain problems. Our technology spans the entire Prime Video catalogue globally, and we enable instant recaps, skip intro timing, ad placement, search, and content moderation.
US, NJ, Newark
At Audible, we believe stories have the power to transform lives. It’s why we work with some of the world’s leading creators to produce and share audio storytelling with our millions of global listeners. We are dreamers and inventors who come from a wide range of backgrounds and experiences to empower and inspire each other. Imagine your future with us. ABOUT THIS ROLE As an Applied Scientist II, you will work on complex problems where neither the problem nor solution is well defined. You'll define and crisply frame research problems while developing novel scientific techniques in domains including machine learning, artificial intelligence (AI), natural language processing (NLP), large language models (LLMs), reinforcement learning (RL), and audio processing. Your primary focus will be on applying and extending existing scientific techniques, as well as inventing new approaches to address specific customer needs and business problems at the project level. You will contribute to internal or external peer-reviewed publications that validate the novelty of your work, while documenting and sharing findings in line with scientific best practices. You will work on LLM applications to enhance Audible's customer experience We work in a highly collaborative environment where you'll primarily influence your team, begin mentoring more junior scientists, and partner with engineers and product managers to implement scalable, efficient approaches for difficult problems. You will operate with some autonomy while knowing when to seek direction to deliver high-quality scientific artifacts. As an Applied Scientist II, you will... - Define and implement scalable, efficient approaches for difficult problems related to audio storytelling and content experiences - Apply and extend state-of-the-art LLM techniques to address specific customer or business needs at the project level - Work on portions of systems, large components, applications, or services supporting machine learning and AI use cases - Apply and extend state-of-the-art techniques in areas like NLP and deep learning to address specific customer or business needs - Execute on team-level goals while creating intellectual property through your work - Apply best practices in software development at the component level, ensuring solutions are testable, reproducible, and efficient - Document and share findings that contribute to the internal and external scientific community - Begin mentoring and developing teammates while gaining experience in tactical work and learning to be strategic - Collaborate with tech and product teams to implement solutions that consider relevant tradeoffs at the component level ABOUT AUDIBLE Audible is the leading producer and provider of audio storytelling. We spark listeners’ imaginations, offering immersive, cinematic experiences full of inspiration and insight to enrich our customers daily lives. We are a global company with an entrepreneurial spirit. We are dreamers and inventors who are passionate about the positive impact Audible can make for our customers and our neighbors. This spirit courses throughout Audible, supporting a culture of creativity and inclusion built on our People Principles and our mission to build more equitable communities in the cities we call home.