Cerebras and National Labs Collaboration Advances Millisecond-Scale Atomic Simulations

May 15, 2024

SUNNYVALE, Calif., May 15, 2024 — Cerebras Systems, in collaboration with researchers from Sandia, Lawrence Livermore, and Los Alamos National Laboratories, has achieved a breakthrough in molecular dynamics (MD) simulations. Using the second generation Cerebras Wafer Scale Engine (WSE-2), researchers were able to perform atomic scale simulations at the millisecond scale – 179x faster than what is possible on the world’s leading supercomputer ‘Frontier,’ which is built with 39,000 GPUs.

Existing supercomputers have been limited to simulating materials at the atomic scale at a rate of 2-3 simulated microseconds per month, restricting our understanding of how materials evolve and behave over longer periods. This breakthrough achieved by Cerebras and its collaborators at national laboratories has shattered this barrier. By harnessing the power of the Cerebras WSE-2, the processor at the heart of the Cerebras CS-2 system, researchers can now simulate materials for milliseconds – an astounding leap that opens up entirely new vistas in materials science.

“The partnership between the NNSA laboratories and Cerebras Systems is part of the Advanced Memory Technology (AMT) program, which aims to accelerate exascale supercomputers by 40x as early as 2025. With Cerebras’ currently deployed wafer-scale computers, the teams achieved this materials science breakthrough and a speedup that exceeded the goal of the AMT program by more than 4X,” said James H. Laros III, Distinguished Member of Technical Staff at Sandia National Laboratories and AMT program lead. “This experience bodes well for future impacts to our program and potential scientific advances.”

The research team innovated across both hardware and software to overcome the limitations of today’s supercomputers. By mapping individual atoms onto the WSE-2’s nearly one million cores and enabling efficient communication between neighboring cores, the system simulated 270,000 timesteps per second across 800,000 atoms – a staggering 179-fold speedup over Frontier, the world’s leading supercomputer. This breakthrough allows researchers to gain unprecedented insights into the long-term behavior and future evolution of materials at the atomic scale.

“This work changes the landscape of what is possible with molecular dynamics simulations,” said Michael James, Chief Architect of Advanced Technologies and co-founder of Cerebras Systems. “Simulations that would have taken a year on a traditional supercomputer can now be completed in just two days. Scientists will now be able to explore previously inaccessible phenomena across a wide range of domains.”

Long timescale simulations will allow scientists to explore previously inaccessible phenomena across a wide range of domains:

  • Materials scientists can now study the long-term behavior of complex materials, such as the evolution of grain boundaries in metals, leading to the development of stronger, more resilient materials.
  • Pharmaceutical researchers can simulate protein folding and drug-target interactions over physiologically relevant timescales, accelerating the discovery of life-saving therapies.
  • Renewable energy experts can optimize catalytic reactions and design more efficient energy storage systems by simulating atomic-scale processes over extended durations.

To achieve this feat, the Cerebras team employed a novel mapping scheme that assigns each atom to a single core on the WSE-2. The cores are organized in a 2D grid, with the physical simulation domain mapped onto this grid to preserve locality. Cores communicate with their neighbors to exchange atom information, allowing for efficient parallel processing. The WSE-2’s unique architecture, with its high memory bandwidth and low-latency communication fabric, enables this fine-grained parallelism to be exploited effectively.

“The NNSA’s Advanced Memory Technologies program started 1.5 years ago with the goal of 40x performance improvement on critical NNSA applications over what can be achieved on exascale systems. We all had our doubts on achieving this goal within the short timeframe, but Cerebras’s technology and team has helped us exceed this goal by demonstrating unprecedented 179x performance improvement on MD simulations,” Siva Rajamanickam, Sandia National Laboratories, Principal Member of Technical Staff. “These results open up new opportunities for materials research and science discoveries beyond what we envisioned. We are excited to continue this collaboration with Cerebras and explore new frontiers in science.”

As Cerebras continues to push the boundaries of high-performance computing with its wafer-scale technology, even more groundbreaking advancements are on the horizon.

“I have been working in atomistic simulation of materials for more than 20 years. During that time, I have participated in massive improvements in both the size and accuracy of the simulations. However, despite all this, we have been unable to increase the actual simulation rate. The wall-clock time required to run simulations has barely budged in the last 15 years,” Aidan Thompson, Sandia National Laboratories, Distinguished Member of Technical Staff. “With the Cerebras Wafer-Scale Engine, we can all of a sudden drive at hypersonic speeds. This joint DOE-Cerebras team has accomplished something unprecedented and highly disruptive. I look forward to seeing how this transforms all kinds of scientific research in the near future.”

For more information, please see https://arxiv.org/abs/2405.07898.

About Cerebras Systems

Cerebras Systems is a team of pioneering computer architects, computer scientists, deep learning researchers, and engineers of all types. We have come together to accelerate generative AI by building from the ground up a new class of AI supercomputer. Our flagship product, the CS-3 system, is powered by the world’s largest and fastest AI processor, our Wafer-Scale Engine-3. CS-3s are quickly and easily clustered together to make the largest AI supercomputers in the world, and make placing models on the supercomputers dead simple by avoiding the complexity of distributed computing. Leading corporations, research institutions, and governments use Cerebras solutions for the development of pathbreaking proprietary models, and to train open-source models with millions of downloads. Cerebras solutions are available through the Cerebras Cloud and on premise.


Source: Cerebras Systems

Subscribe to HPCwire's Weekly Update!

Be the most informed person in the room! Stay ahead of the tech trends with industry updates delivered to you every week!

ISC 2024 Takeaways: Love for Top500, Extending HPC Systems, and Media Bashing

May 23, 2024

The ISC High Performance show is typically about time-to-science, but breakout sessions also focused on Europe's tech sovereignty, server infrastructure, storage, throughput, and new computing technologies. This round Read more…

HPC Pioneer Gordon Bell Passed Away

May 22, 2024

Legendary computer scientist Gordon Bell passed away last Friday at his home in Coronado, CA. He was 89. The New York Times has a nice tribute piece. A long-time pioneer with Digital Equipment Corp, he pushed hard for de Read more…

ISC 2024 — A Few Quantum Gems and Slides from a Packed QC Agenda

May 22, 2024

If you were looking for quantum computing content, ISC 2024 was a good place to be last week — there were around 20 quantum computing related sessions. QC even earned a slide in Kathy Yelick’s opening keynote — Bey Read more…

Atos Outlines Plans to Get Acquired, and a Path Forward

May 21, 2024

Atos – via its subsidiary Eviden – is the second major supercomputer maker outside of HPE, while others have largely dropped out. The lack of integrators and Atos' financial turmoil have the HPC market worried. If Atos goes under, HPE will be the only major option for building large-scale systems. Read more…

Core42 Is Building Its 172 Million-core AI Supercomputer in Texas

May 20, 2024

UAE-based Core42 is building an AI supercomputer with 172 million cores which will become operational later this year. The system, Condor Galaxy 3, was announced earlier this year and will have 192 nodes with Cerebras Read more…

Google Announces Sixth-generation AI Chip, a TPU Called Trillium

May 17, 2024

On Tuesday May 14th, Google announced its sixth-generation TPU (tensor processing unit) called Trillium.  The chip, essentially a TPU v6, is the company's latest weapon in the AI battle with GPU maker Nvidia and clou Read more…

ISC 2024 Takeaways: Love for Top500, Extending HPC Systems, and Media Bashing

May 23, 2024

The ISC High Performance show is typically about time-to-science, but breakout sessions also focused on Europe's tech sovereignty, server infrastructure, storag Read more…

ISC 2024 — A Few Quantum Gems and Slides from a Packed QC Agenda

May 22, 2024

If you were looking for quantum computing content, ISC 2024 was a good place to be last week — there were around 20 quantum computing related sessions. QC eve Read more…

Atos Outlines Plans to Get Acquired, and a Path Forward

May 21, 2024

Atos – via its subsidiary Eviden – is the second major supercomputer maker outside of HPE, while others have largely dropped out. The lack of integrators and Atos' financial turmoil have the HPC market worried. If Atos goes under, HPE will be the only major option for building large-scale systems. Read more…

Google Announces Sixth-generation AI Chip, a TPU Called Trillium

May 17, 2024

On Tuesday May 14th, Google announced its sixth-generation TPU (tensor processing unit) called Trillium.  The chip, essentially a TPU v6, is the company's l Read more…

Europe’s Race towards Quantum-HPC Integration and Quantum Advantage

May 16, 2024

What an interesting panel, Quantum Advantage — Where are We and What is Needed? While the panelists looked slightly weary — their’s was, after all, one of Read more…

The Future of AI in Science

May 15, 2024

AI is one of the most transformative and valuable scientific tools ever developed. By harnessing vast amounts of data and computational power, AI systems can un Read more…

Some Reasons Why Aurora Didn’t Take First Place in the Top500 List

May 15, 2024

The makers of the Aurora supercomputer, which is housed at the Argonne National Laboratory, gave some reasons why the system didn't make the top spot on the Top Read more…

ISC 2024 Keynote: High-precision Computing Will Be a Foundation for AI Models

May 15, 2024

Some scientific computing applications cannot sacrifice accuracy and will always require high-precision computing. Therefore, conventional high-performance c Read more…

Synopsys Eats Ansys: Does HPC Get Indigestion?

February 8, 2024

Recently, it was announced that Synopsys is buying HPC tool developer Ansys. Started in Pittsburgh, Pa., in 1970 as Swanson Analysis Systems, Inc. (SASI) by John Swanson (and eventually renamed), Ansys serves the CAE (Computer Aided Engineering)/multiphysics engineering simulation market. Read more…

Nvidia H100: Are 550,000 GPUs Enough for This Year?

August 17, 2023

The GPU Squeeze continues to place a premium on Nvidia H100 GPUs. In a recent Financial Times article, Nvidia reports that it expects to ship 550,000 of its lat Read more…

Comparing NVIDIA A100 and NVIDIA L40S: Which GPU is Ideal for AI and Graphics-Intensive Workloads?

October 30, 2023

With long lead times for the NVIDIA H100 and A100 GPUs, many organizations are looking at the new NVIDIA L40S GPU, which it’s a new GPU optimized for AI and g Read more…

Atos Outlines Plans to Get Acquired, and a Path Forward

May 21, 2024

Atos – via its subsidiary Eviden – is the second major supercomputer maker outside of HPE, while others have largely dropped out. The lack of integrators and Atos' financial turmoil have the HPC market worried. If Atos goes under, HPE will be the only major option for building large-scale systems. Read more…

Choosing the Right GPU for LLM Inference and Training

December 11, 2023

Accelerating the training and inference processes of deep learning models is crucial for unleashing their true potential and NVIDIA GPUs have emerged as a game- Read more…

AMD MI3000A

How AMD May Get Across the CUDA Moat

October 5, 2023

When discussing GenAI, the term "GPU" almost always enters the conversation and the topic often moves toward performance and access. Interestingly, the word "GPU" is assumed to mean "Nvidia" products. (As an aside, the popular Nvidia hardware used in GenAI are not technically... Read more…

Nvidia’s New Blackwell GPU Can Train AI Models with Trillions of Parameters

March 18, 2024

Nvidia's latest and fastest GPU, codenamed Blackwell, is here and will underpin the company's AI plans this year. The chip offers performance improvements from Read more…

Some Reasons Why Aurora Didn’t Take First Place in the Top500 List

May 15, 2024

The makers of the Aurora supercomputer, which is housed at the Argonne National Laboratory, gave some reasons why the system didn't make the top spot on the Top Read more…

Leading Solution Providers

Contributors

Shutterstock 1606064203

Meta’s Zuckerberg Puts Its AI Future in the Hands of 600,000 GPUs

January 25, 2024

In under two minutes, Meta's CEO, Mark Zuckerberg, laid out the company's AI plans, which included a plan to build an artificial intelligence system with the eq Read more…

Eyes on the Quantum Prize – D-Wave Says its Time is Now

January 30, 2024

Early quantum computing pioneer D-Wave again asserted – that at least for D-Wave – the commercial quantum era has begun. Speaking at its first in-person Ana Read more…

The GenAI Datacenter Squeeze Is Here

February 1, 2024

The immediate effect of the GenAI GPU Squeeze was to reduce availability, either direct purchase or cloud access, increase cost, and push demand through the roof. A secondary issue has been developing over the last several years. Even though your organization secured several racks... Read more…

Shutterstock 1285747942

AMD’s Horsepower-packed MI300X GPU Beats Nvidia’s Upcoming H200

December 7, 2023

AMD and Nvidia are locked in an AI performance battle – much like the gaming GPU performance clash the companies have waged for decades. AMD has claimed it Read more…

The NASA Black Hole Plunge

May 7, 2024

We have all thought about it. No one has done it, but now, thanks to HPC, we see what it looks like. Hold on to your feet because NASA has released videos of wh Read more…

Intel Plans Falcon Shores 2 GPU Supercomputing Chip for 2026  

August 8, 2023

Intel is planning to onboard a new version of the Falcon Shores chip in 2026, which is code-named Falcon Shores 2. The new product was announced by CEO Pat Gel Read more…

GenAI Having Major Impact on Data Culture, Survey Says

February 21, 2024

While 2023 was the year of GenAI, the adoption rates for GenAI did not match expectations. Most organizations are continuing to invest in GenAI but are yet to Read more…

How the Chip Industry is Helping a Battery Company

May 8, 2024

Chip companies, once seen as engineering pure plays, are now at the center of geopolitical intrigue. Chip manufacturing firms, especially TSMC and Intel, have b Read more…

  • arrow
  • Click Here for More Headlines
  • arrow
HPCwire