Alex-Bayen.jpg
Alexandre Bayen is the Liao-Cho Professor of Engineering at the University of California Berkeley and director of its Institute of Transportation Studies. Bayen plays leading roles in multiple transportation projects.
Courtesy of Alexandre Bayen

Alexandre Bayen is a driving force behind mixed-autonomy traffic

Coordinated automation could improve traffic flow, boost efficiency, and slash emissions. A combination of machine learning, big data, and Amazon Web Services is making this future possible.

The smooth-flowing traffic of the future is just around the corner. Advances in vehicle automation are converging with developments in machine learning (ML) and cloud computing to create self-driving vehicles that not only control themselves safely, but also have an oversized beneficial effect on the journeys of all the regular drivers on the road around them. Welcome to “mixed autonomy traffic”.

Leading the pack into this future is Alexandre Bayen, the Liao-Cho Professor of Engineering at the University of California Berkeley and director of its Institute of Transportation Studies. An expert in control and optimization, Bayen is playing leading roles in multiple transportation projects, ranging from cutting-edge, open-source traffic simulation and optimization, to large scale freeway observation that involves putting automated vehicles into real traffic to explore the impact of ML-derived self-driving behaviors. These automated vehicles also have human supervisors at the wheel, ready to take over the vehicle at any time if needed.

Before delving into Bayen’s work, an example of the promise of mixed autonomy traffic is in order.

Traffic jam experiment
This video is from a 2008 experiment in which people are attempting to maintain the same speed while driving single-file around a circular track.

Anyone regularly caught in “phantom” traffic jams, which have no obvious cause, knows how annoying they are. It is simply the nature of human drivers to create these so-called “stop-and-go waves” — we just can’t help jamming up then spreading out on the road, as illustrated by a brief video (above) of a classic 2008 experiment in which people are attempting to maintain the same speed while driving single-file around a circular track.

Fast forward to 2017, to a series of similar experiments led by Bayen’s collaborators, Jonathan Sprinkle of the University of Arizona and Daniel Work of Vanderbilt University. This work echoed the 2008 experiment, but with an enormous difference: of the 20 or so cars on a circular track, one of them could switch into self-driving mode. When it did, the effect on the stop-and-go waves was immediate — and remarkable.

Self-driving cars experiment demonstrates dramatic improvements in traffic flow

Simply through the slowing or accelerating of this single car, in accordance with its traffic-optimization algorithms, the traffic waves dissipated significantly. In one test, fuel consumption of the cars in the ring was reduced by more than 40% and excessive braking events dropped from 8.5 per vehicle-kilometer to near zero.

The experimenters concluded that traffic flow control would be possible in real-life traffic with less than 5% of cars being automated.

A self-driving future

With that in mind, what will happen to our existing traffic flow when increasing numbers of vehicles are self-driving? This is the future being shaped by Bayen and his group. At the center of his work is an open-source framework called FLOW. With deep reinforcement learning at its heart, FLOW is an optimization and microsimulation tool for traffic flow. Don’t be fooled by “micro” in this context — the simulation features hundreds of thousands of vehicles on complex road systems. FLOW allows the virtual exploration of complex traffic optimization challenges on a wide variety of road set-ups.

“Traffic simulation engines have become really good, very accurate, in the last decade. And the computation required has become really tractable, mostly because of scalable cloud computing offered by Amazon Web Services and others,” says Bayen.

Deep reinforcement learning is particularly suited to developing mixed-autonomy traffic optimization because it enables simulated self-driving vehicles to try out different driving behaviors. If a set of driving policies results in lower fuel use without compromising journey time, for example, the algorithm is rewarded. “Ten years ago it was really hard to compute the outcome of experiments in simulation — and very costly. You could do a couple of intersections, and maybe a couple hundred vehicles,” says Bayen. “With the plethora of data available now, combined with the ability to do these computations very fast, it has become really quick to compute the rewards and to iterate until you get something that works very well.”

Achieving a FLOW state

Bayen is keen to clarify the primary goal of FLOW. “It’s important to differentiate between boosting energy efficiency and reducing congestion. We are not attempting to fix congestion — that is not our goal, and these would not be the right tools. We are improving the energy efficiency of traffic, which is a very different problem.”

We are not attempting to fix congestion — that is not our goal, and these would not be the right tools. We are improving the energy efficiency of traffic, which is a very different problem.
Alexandre Bayen

Indeed, in simulations, FLOW’s algorithms have a minimal effect on travel time — but a dramatic effect on the driving experience, Bayen explains. “The amount of braking is significantly reduced and the amount of acceleration — where most of the energy is burned and pollutants emitted — has been significantly reduced as well. That's the main challenge.”

In 2019, Bayen received an Amazon ML Research Award to support the development of "Applications of Deep-RL for Training Connected, Autonomous Vehicles in Mixed Environments". But even before the award, FLOW was intrinsically linked to Amazon Web Services (AWS), Bayen explains. “When we started FLOW in 2018, there were only three tools widely used for microsimulation of traffic: SUMO, Aimsun, and PTV Vissim. SUMO was an open-source platform already running on AWS, but Aimsun — now owned by Siemens Mobility — built the first instantiation of their software on the AWS cloud specifically for us,” says Bayen. “The FLOW Project was the first time anyone managed to put these three big components together: the machine learning, the cloud computing, and the simulation engine. It was historic.”

A key reason this combination is important, Sprinkle says, is big data: “For societal-scale systems to take advantage of ML, they need to take advantage of these gigantic datasets. Hosting the ML algorithms on AWS — in the same place the data are — speeds up discovery.”

The success of FLOW generated a lot of interest in Bayen’s group, including from the US government, which subsequently decided to fund the research. That is when Bayen and a broad collaboration, called the CIRCLES Consortium, was formed, with Bayen, Work, and Sprinkle among the co-principal investigators. They started working with Toyota, GM, and Nissan, to develop a proof-of-concept to demonstrate that mixed-autonomy traffic control actually works on the road. “That is what we are doing now, with the generous funding of the US Department of Energy,” says Bayen.

Part of this effort is a project called I-24 Mobility Technology Interstate Observation Network (I-24 MOTION). The CIRCLES Consortium is installing video monitoring infrastructure along six miles of I-24 in Tennessee, to gather extensive, top-quality traffic data. When completed in 2022, it will consist of 400 pole-mounted, 4k-resolution cameras. “The network is already gathering an astronomical amount of data — on the order of petabytes,” says Bayen. “It will not only provide the Tennessee Department of Transportation with a lot more operational capabilities for freeway operations, but also provide the research community with an unprecedented data set that has the potential to unveil a lot of interesting traffic features.”

Real life traffic testing

This is where the rubber hits the road. This year, the CIRCLES Consortium is deploying self-driving vehicles on that same stretch of I-24, to see how ML-derived self-driving algorithms might positively impact real-world traffic. “We’re hoping that by driving a few cars differently, it will reduce energy use for the entire stream of traffic,” says Sprinkle.

Heavy morning traffic on Highway 101 going through Silicon Valley, South San Francisco Bay Area
Alexandre Bayen says going from simulations to real-world deployment is significant. “If something runs really well in simulation, one still needs to be certain that it will transfer well to hardware and run well with real cars on real roads using imperfect data."
Sundry Photography/Getty Images

“This summer, we're doing 14 vehicles — four with automation and 10 as monitoring vehicles gathering local measurements,” says Bayen. Next year, another live deployment is planned, but with a dramatic increase in the number of automated and monitoring vehicles. 

This step from simulation to real-world deployment is more like a giant leap. “If something runs really well in simulation, one still needs to be certain that it will transfer well to hardware and run well with real cars on real roads using imperfect data. That's a big challenge,” says Bayen.

To that end, since 2016, the US National Science Foundation has funded efforts to develop the software framework that enables FLOW to be deployed on a variety of real vehicles and many different hardware platforms. The real-world deployment is a cautious, painstaking process. “We have facilities at Berkeley and Vanderbilt for low speed, and later high-speed testing, that enables us to work through the sequence of steps,” Bayen notes. “Now we’ve done this on private roads, open roads, and have progressed to freeway traffic.”  

Another challenge for this field is predicting how cars might transmit their locations in the future. There are also ongoing debates around how driver movement data will or should be collected, protected, transmitted, and shared, says Bayen. “Our job is to work on the different architectures that can support these many potential paradigms. These include fully ‘decentralized’ vehicles that do not need to talk to each other or to a central authority to improve overall traffic flow, or fully centralized, in which everybody talks to everybody. Or partially coordinated, in which cars only talk to their neighboring cars, and so on. While we wait for a public policy on this, we are developing an entire portfolio of algorithms spanning a multitude of paradigms. It's a lot of work!”

But it is work worth doing, says Bayen, because FLOW is highly scalable. “Many cities have good models of their traffic systems. Putting our software on top of it is really not difficult if those models run in AIMSUN or SUMO, two of the three major simulators. We can put such models into our framework and apply machine learning directly to it.” The cloud-based aspect is essential to this scalability. “Before the cloud became a reality in this arena, people would have a specific architecture that their traffic models would run on. But because FLOW is open source and on AWS, anyone can run it, from anywhere, including other research groups. That’s the power of the cloud.”

Work agrees: “Providing an open-source approach empowers new researchers to explore their own ideas. And using machine learning for large-scale systems is exciting because of the potential for benefits to all — even if only a few parts of the system change their behavior.” And the benefits also extend to the local and global environment, says Bayen, because the emissions per vehicle — both direct, and indirect for electric vehicles — are likely to be significantly reduced.

With the rate at which the technology of mixed-autonomy traffic is advancing, the generation of drivers hitting the roads five years from now may be confused when their parents marvel at how smooth freeway traffic is “these days”, despite the large numbers of vehicles on the road. For the rest of us, knowing that phantom jams’ days are numbered will probably make them easier to bear. Honk if you disagree.

Related content

US, WA, Seattle
The Automated Reasoning Group in the Amazon Neuron team is looking for an Applied Scientist to work on the intersection of Artificial Intelligence and program analysis to raise the code quality bar in our state-of-the-art deep learning compiler stack. This stack is designed to optimize application models across diverse domains, including Large Language and Vision, originating from leading frameworks such as PyTorch and JAX. Your role will involve working closely with our custom-built Machine Learning accelerator, Trainium, which represents the forefront of innovation for advanced ML capabilities, and is the underpinning of Generative AI. In this role as an Applied Scientist, you'll be instrumental in designing, developing, and deploying analyzers for ML compiler stages and compiler IRs. You will architect and implement business-critical tooling, publish research, and mentor a brilliant team of experienced scientists and engineers. You will need to be technically capable, credible, and curious in your own right as a trusted AWS Neuron engineer, innovating on behalf of our customers. Your responsibilities will involve tackling crucial challenges alongside a talented engineering team, contributing to leading-edge design and research in compiler technology and deep-learning systems software. Strong experience in programming languages, compilers, program analyzers, theorem provers, and program synthesis engines will be a benefit in this role. A background in machine learning and AI accelerators is preferred but not required.
US, CA, San Francisco
Join the next revolution in robotics at Amazon's Frontier AI & Robotics team, where you'll work alongside world-renowned AI pioneers to push the boundaries of what's possible in robotic intelligence. As a Member of Technical Staff, you'll be at the forefront of developing breakthrough foundation models that enable robots to perceive, understand, and interact with the world in unprecedented ways. You'll drive independent research initiatives in areas such as perception, manipulation, science understanding, locomotion, manipulation, sim2real transfer, multi-modal foundation models and multi-task robot learning, designing novel frameworks that bridge the gap between state-of-the-art research and real-world deployment at Amazon scale. In this role, you'll balance innovative technical exploration with practical implementation, collaborating with platform teams to ensure your models and algorithms perform robustly in dynamic real-world environments. You'll have access to Amazon's vast computational resources, enabling you to tackle ambitious problems in areas like very large multi-modal robotic foundation models and efficient, promptable model architectures that can scale across diverse robotic applications. Key job responsibilities - Drive independent research initiatives across the robotics stack, including robotics foundation models, focusing on breakthrough approaches in perception, and manipulation, for example open-vocabulary panoptic scene understanding, scaling up multi-modal LLMs, sim2real/real2sim techniques, end-to-end vision-language-action models, efficient model inference, video tokenization - Design and implement novel deep learning architectures that push the boundaries of what robots can understand and accomplish - Lead full-stack robotics projects from conceptualization through deployment, taking a system-level approach that integrates hardware considerations with algorithmic development, ensuring robust performance in production environments - Collaborate with platform and hardware teams to ensure seamless integration across the entire robotics stack, optimizing and scaling models for real-world applications - Contribute to the team's technical strategy and help shape our approach to next-generation robotics challenges A day in the life - Design and implement novel foundation model architectures and innovative systems and algorithms, leveraging our extensive infrastructure to prototype and evaluate at scale - Collaborate with our world-class research team to solve complex technical challenges - Lead technical initiatives from conception to deployment, working closely with robotics engineers to integrate your solutions into production systems - Participate in technical discussions and brainstorming sessions with team leaders and fellow scientists - Leverage our massive compute cluster and extensive robotics infrastructure to rapidly prototype and validate new ideas - Transform theoretical insights into practical solutions that can handle the complexities of real-world robotics applications About the team At Frontier AI & Robotics, we're not just advancing robotics – we're reimagining it from the ground up. Our team is building the future of intelligent robotics through innovative foundation models and end-to-end learned systems. We tackle some of the most challenging problems in AI and robotics, from developing sophisticated perception systems to creating adaptive manipulation strategies that work in complex, real-world scenarios. What sets us apart is our unique combination of ambitious research vision and practical impact. We leverage Amazon's massive computational infrastructure and rich real-world datasets to train and deploy state-of-the-art foundation models. Our work spans the full spectrum of robotics intelligence – from multimodal perception using images, videos, and sensor data, to sophisticated manipulation strategies that can handle diverse real-world scenarios. We're building systems that don't just work in the lab, but scale to meet the demands of Amazon's global operations. Join us if you're excited about pushing the boundaries of what's possible in robotics, working with world-class researchers, and seeing your innovations deployed at unprecedented scale.
US, CA, San Francisco
Join the next revolution in robotics at Amazon's Frontier AI & Robotics team, where you'll work alongside world-renowned AI pioneers to push the boundaries of what's possible in robotic intelligence. As a Member of Technical Staff, you'll be at the forefront of developing breakthrough foundation models that enable robots to perceive, understand, and interact with the world in unprecedented ways. You'll drive independent research initiatives in areas such as perception, manipulation, science understanding, locomotion, manipulation, sim2real transfer, multi-modal foundation models and multi-task robot learning, designing novel frameworks that bridge the gap between state-of-the-art research and real-world deployment at Amazon scale. In this role, you'll balance innovative technical exploration with practical implementation, collaborating with platform teams to ensure your models and algorithms perform robustly in dynamic real-world environments. You'll have access to Amazon's vast computational resources, enabling you to tackle ambitious problems in areas like very large multi-modal robotic foundation models and efficient, promptable model architectures that can scale across diverse robotic applications. Key job responsibilities - Drive independent research initiatives across the robotics stack, including robotics foundation models, focusing on breakthrough approaches in perception, and manipulation, for example open-vocabulary panoptic scene understanding, scaling up multi-modal LLMs, sim2real/real2sim techniques, end-to-end vision-language-action models, efficient model inference, video tokenization - Design and implement novel deep learning architectures that push the boundaries of what robots can understand and accomplish - Lead full-stack robotics projects from conceptualization through deployment, taking a system-level approach that integrates hardware considerations with algorithmic development, ensuring robust performance in production environments - Collaborate with platform and hardware teams to ensure seamless integration across the entire robotics stack, optimizing and scaling models for real-world applications - Contribute to the team's technical strategy and help shape our approach to next-generation robotics challenges A day in the life - Design and implement novel foundation model architectures and innovative systems and algorithms, leveraging our extensive infrastructure to prototype and evaluate at scale - Collaborate with our world-class research team to solve complex technical challenges - Lead technical initiatives from conception to deployment, working closely with robotics engineers to integrate your solutions into production systems - Participate in technical discussions and brainstorming sessions with team leaders and fellow scientists - Leverage our massive compute cluster and extensive robotics infrastructure to rapidly prototype and validate new ideas - Transform theoretical insights into practical solutions that can handle the complexities of real-world robotics applications About the team At Frontier AI & Robotics, we're not just advancing robotics – we're reimagining it from the ground up. Our team is building the future of intelligent robotics through innovative foundation models and end-to-end learned systems. We tackle some of the most challenging problems in AI and robotics, from developing sophisticated perception systems to creating adaptive manipulation strategies that work in complex, real-world scenarios. What sets us apart is our unique combination of ambitious research vision and practical impact. We leverage Amazon's massive computational infrastructure and rich real-world datasets to train and deploy state-of-the-art foundation models. Our work spans the full spectrum of robotics intelligence – from multimodal perception using images, videos, and sensor data, to sophisticated manipulation strategies that can handle diverse real-world scenarios. We're building systems that don't just work in the lab, but scale to meet the demands of Amazon's global operations. Join us if you're excited about pushing the boundaries of what's possible in robotics, working with world-class researchers, and seeing your innovations deployed at unprecedented scale.
US, CA, Palo Alto
The Amazon Search team creates customer-focused search solutions and technologies. Whenever a customer visits an Amazon site worldwide and types in a query or browses through product categories, Amazon Product Search services go to work. We design, develop, and deploy high-performance distributed search systems that rank a catalog of billions of products for hundreds of millions of shoppers. The Search Relevance team owns the ranking models that decide the order of results on every Amazon search page. In this role, you will design and post-train deep ranking models, including LLM-based rankers and multi-tower deep learning models, that jointly optimize purchase, relevance, and personalization. You will invent modeling and training techniques that push the Pareto frontier across multiple objectives, and take your work end to end from novel research prototype through offline evaluation to production online experimentation. Personalization is a first-class objective on this team. You will build models that reason over each customer's history, durable preferences, and query intent to decide which results best fit that specific customer, rather than optimizing a single population-level ranking. We treat search as an active research frontier and invest heavily in staying at the leading edge of ML. Beyond today's ranking stack, our current explorations include LLM agents that reason and plan across multi-step workflows, tool-augmented foundation models, and new paradigms that combine retrieval, reasoning, and personalization. You will help chart where search goes next, and see your ideas ship to real customers within weeks, not quarters. You will work in a dynamic, entrepreneurial team while leveraging the resources of Amazon.com, one of the world's leading technology companies. Please visit https://www.amazon.science for more information. Key job responsibilities Your responsibilities include but are not limited to: - Design, train, and deploy state-of-the-art ranking models that decide how results are ordered on Amazon search, spanning LLM-based rankers and multi-tower deep learning architectures that jointly model engagement, relevance, and personalization. - Post-train LLMs and ranking models with supervised fine-tuning, reinforcement learning (e.g. GRPO, DPO, RLHF), knowledge distillation, and listwise ranking losses (e.g. LambdaLoss, ListNet, ListMLE). - Compose multiple objectives (engagement, relevance, personalization) into a single ranking through principled multi-objective optimization at inference. - Design large-scale label pipelines, including LLM-as-teacher supervision, that turn customer signals and expert judgment into training and reward signals. - Optimize inference for production ranking models through quantization, quantization-aware training, teacher-student distillation, and serving-stack tuning. - Evaluate proposed solutions through offline benchmarks and online A/B tests, and drive the analysis that decides whether a change ships. - Publish and present your work at internal and external scientific venues in ML, NLP, and IR.
CN, 31, Shanghai
Team & Project Overview The NBS Data Central team powers analytics, data science, and AI capabilities for Worldwide Global Selling (WWGS). We build scalable data products, and insight-generation systems that drive seller growth across 10+ marketplaces. Seller Intelligence is a P0 foundation theme at the Global Selling level, formed by merging "One Tagging" and "Good Contact" workstreams. It provides seller identity, segmentation, and contact-reach infrastructure that underpins all downstream seller-facing AI workflows — including intelligent outreach, personalized recommendations, and automated engagement. Scope of Impact Own the science pillar for Seller Intelligence within a cross-functional POD (PM + DE + DS + SDE) Directly impact seller engagement metrics across CN, IN, LATAM, and East-Asia expansion regions Models and data products consumed by 5+ downstream teams (ESM, NSR, MKT, NBS AI Ops, ROC) Influence $100M+ annual seller GMS through improved segmentation and contact optimization Key job responsibilities Design and deliver seller segmentation and propensity models at scale — incorporating GMS, category, growth trajectory, engagement signals, and lifecycle stage. Build contact quality scoring and lifecycle management systems (coverage optimization, dormancy detection, reactivation modeling). Define success metrics, experimentation frameworks (A/B, causal inference), and measurement methodology for seller engagement interventions. Productionize ML models and data products — partner with engineering to deploy seller scores, contact quality indices, and recommendation signals. Explore LLM/GenAI applications: automated insight generation from seller data, contact intent classification, and intelligent report synthesis. Serve as the science representative in bi-weekly NBS theme reviews; present findings and proposals to theme Bar Raisers and leadership. Collaborate with BIE team members to democratize analytical outputs via dashboards and self-serve tools. Contribute to cross-marketplace seller behavior analysis supporting Global Expansion strategy (IN, KR, VN, LATAM). Evaluate, integrate, and iterate on AI systems — assess new AI/ML tools, frameworks, and third-party models for applicability to seller intelligence use cases.
GB, London
Come build the future of entertainment with us. Are you interested in shaping the future of movies and television? Do you want to define the next generation of how and what Amazon customers are watching? Prime Video is a premium streaming service that offers customers a vast collection of TV shows and movies — all with the ease of finding what they love to watch in one place. We offer customers thousands of popular movies and TV shows from Originals and Exclusive content to exciting live sports events. We also offer our members the opportunity to subscribe to add-on channels which they can cancel at anytime and to rent or buy new release movies and TV box sets on the Prime Video Store. Prime Video is a fast-paced, growth business — available in over 240 countries and territories worldwide. The team works in a dynamic environment where innovating on behalf of our customers is at the heart of everything we do. If this sounds exciting to you, please read on. Prime Video Commerce's mission is to present the right offer to the right customer at the right time — across subscriptions, channels, and transactional video, in every market and on every device. Our science team replaces static business rules with ML-driven decisions that personalise the entire commerce journey, from discovery through checkout and beyond. We operate at scale across hundreds of millions of customers, and we are expanding into new frontiers — combining the latest advances in agentic and generative AI, behavioural simulation, and causal inference to understand the impact of our decisions before they reach customers. We are looking for an Applied Scientist to join the Prime Video Commerce Insights team in London. You will develop and deploy customer-facing models, understand customer behaviour at scale, and explore emerging techniques that help us make better decisions faster. This is a delivery focused role within a high-visibility multidisciplinary group of engineers and scientists, focused on improving the customer experience for Prime Video. Key job responsibilities - Research, design, and implement machine learning approaches (e.g. reinforcement learning and recommendation systems) that personalise across different customer touch points. - Collaborate with engineers to deploy and integrate successful experiment results into large-scale, complex Amazon production systems with low latency. - Design and execute rigorous experiments to demonstrate the technical efficacy and business value of your methods. - Act as a subject-matter expert and help define the science roadmap and research agenda in line with organisational priorities and production constraints. - Provide machine learning thought leadership to technical and business leaders, thinking strategically about business, product, and technical challenges. - Work with technical product managers to work backwards from what matters to customers and deliver ML-backed solutions. - Share results with the team and wider scientific community through documents that are both statistically rigorous and compellingly relevant. A day in the life You will be a research leader and innovator within the Commerce Insights organisation. You will collaborate with talented engineers and senior leaders to solve problems that are uniquely challenging at Amazon's scale: personalising commerce decisions across multiple business lines, balancing competing objectives, and positively impacting hundreds of millions of customers worldwide. The problems here are technically deep — combining large-scale ML, causal reasoning, and behavioural modelling in a domain where every decision carries real revenue and customer-experience consequences. Your research will ship to production and move metrics that matter. About the team You will join a team of engineers and applied scientists with a proven track record of solving highly complex, ambiguous problems — work that has produced patents and publications at top-tier conferences. The team has direct visibility to senior Prime Video leadership and collaborates broadly across Commerce, Content, and Platform teams to shape how customers discover, subscribe to, and engage with video content. This is a team that operates at the intersection of rigorous research and real-world impact, where your ideas move from whiteboard to production for hundreds of millions of customers.
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 What will you do? - 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.
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 What will you do? - 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.
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 What will you do? - 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, MA, Boston
Are you excited about applying machine learning and applied mathematics to real-world systems at massive scale? As an Applied Scientist on this newly formed team, you will collaborate closely with scientists and engineers to bring research into production across a broad portfolio of problems — from computer vision perception platforms to building-wide optimization and orchestration. You will frame ambiguous business problems as tractable scientific challenges and implement novel machine learning (ML) systems, first-principles models, embedded systems prototypes, and performance optimizations in both prototype and production environments. This is a ground-floor opportunity to shape the scientific direction of a new organization, where your contributions will directly influence how Amazon's fulfillment network operates and evolves. Key job responsibilities - Design, develop, and deploy ML and scientific solutions spanning classical machine learning, statistical modeling, computer vision, optimization, and physics-informed modeling in production environments. - Rapidly ramp on unfamiliar problem domains, frame ambiguous business problems as tractable scientific challenges, and prototype solutions end to end. - Author or co-author research findings for internal or external peer-reviewed venues, and provide peer feedback on research procedures and results across teams. - Prototype and evaluate sensing hardware and lightweight, edge-deployable models that run on commodity compute under real-world constraints. - Collaborate across multiple science and engineering teams to integrate your solutions into deployment architecture, mentoring less experienced scientists along the way. A day in the life You might start your morning reviewing experiment results from an overnight model training run, then shift into a design discussion with engineers on how to deploy a new computer vision model to edge hardware in a fulfillment center. After lunch, you could be prototyping a physics-informed optimization approach, writing up findings for a research paper, or pairing with a teammate to debug a tricky data pipeline. As part of a new and growing organization, you will have a direct hand in shaping team practices, scientific roadmaps, and the tools you use every day. About the team Our team sits within Amazon's fulfillment technology organization and applies a range of scientific disciplines — including computer vision, optimization, reinforcement learning, and statistical modeling — to improve how goods move through Amazon's global fulfillment network. We build the models and systems that drive real-time orchestration, optimizing throughput, flow, and operational performance at scale. As a newly formed organization, we are building our culture and scientific agenda from the ground up. You will join a collaborative, inclusive group of scientists and engineers who value experimentation, rigorous research, and delivering measurable impact for customers.