PEARC21 Panel: Wafer-Scale-Engine Technology Accelerates Machine Learning, HPC

By Ken Chiacchia, Pittsburgh Supercomputing Center/XSEDE

July 21, 2021

Early use of Cerebras’ CS-1 server and wafer-scale engine (WSE) has demonstrated promising acceleration of machine-learning algorithms, according to participants in the Scientific Research Enabled by CS-1 Systems panel, presented at the PEARC21 conference. The panel, which for the first time brought together leading teams employing the CS-1 at the Pittsburgh Supercomputing Center (PSC), Argonne National Laboratory (ANL) and Lawrence Livermore National Laboratory (LLNL) charted out the promise of the technology as well as the next steps in applying it to artificial intelligence and HPC projects that do not utilize AI.

The PEARC conference series provides a forum for discussing challenges, opportunities and solutions among the broad range of participants in the research computing community. This community-driven effort builds on successes of the past, and aims to grow and be more inclusive by involving additional local, regional, national and international cyberinfrastructure and research computing partners spanning academia, government and industry. PEARC21, Evolution Across All Dimensions, was offered this year as a virtual event (July 19-22).

Cerebras Systems Technology Summary and Outlook

Moderated by co-organizer Sergiu Sanielevici, director of user support at PSC, the panel kicked off with a presentation by co-organizer Natalia Vassilieva, director of product at Cerebras.

While the field has progressed phenomenally over the past decade, these advances have come at a computational cost, Vassilieva explained. Ballooning memory requirements for training have driven proportional increases in required petaflops-days, with OpenAI’s GPT-3 language model requiring about 116 days to train on 1,024 Nvidia V100 GPUs.

“Modern models need much more compute than can be [fit] on a single processor,” she said, and scaling for distributed training is far from ideal. “As you scale out to multiple devices, at some point you start to observe communication bottlenecks” and other limitations. “We need more compute per device, and the ability to rely less on data parallel training.”

“I think everybody understands that the current approach…is not sustainable,” she said.

The CS-1, and the newly introduced second-generation WSE CS-2, contain 400,000 and 850,000 cores per WSE, providing 18 or 40 GB of memory, respectively, one cycle away from the compute element – a memory bandwidth of 9 or 20 PByte/s. The extreme bandwidth within the chip (100 or 220 Pbit/s) also avoids a bottleneck in conventional systems. By harnessing unprecedented local memory and compute cores, the WSEs offers computational scaling with vastly reduced bottlenecks. The system represents a flexible and dynamic solution to the challenges of fine-grained sparsity and conditional and dynamic machine-learning techniques, Vassilieva said.

The Promise of CS-1/WSE for Research in Science and Engineering

Paola Buitrago, co-organizer and director of AI & Big Data at PSC and principal investigator of the center’s CS-1 based Neocortex, surveyed the state of the art in machine learning – and the challenges the field currently faces. Neocortex explores a slightly different approach to leveraging the CS-1 than other deployments, using an HPE SuperDome Flex server as a single, high-memory CPU intermediary with the user and federation with PSC’s larger Bridges-2 system. The unique high-memory configuration is intended to offer advantages in combined Big Data/AI applications.

Improvements in neural language models’ performance have required more compute and more memory, with the number of parameters of recent transformer-type networks surpassing hundreds of billions or trillions. Generative adversarial networks, domain adaptation and reinforcement learning approaches have also added complexity, Buitrago said.

Nor is the expense of improved ML models limited to computation, she explained. An analysis based on information released by Google estimated the monetary cost of training the 175-billion parameter GPT-3 at about $10 million. At that scale, reducing ImageNet error rate from 11.5 percent to 1 percent would represent $100 billion billion – $1020 – she added.

“As models increase in size and the compute requirements increase, [we] also find that to further improve the models’ performance … becomes prohibitive with existing approaches,” Buitrago said. “The field is calling for a change in paradigm…CS-1, certainly as it was conceived, proposes a different approach to machine-learning training” by offering ways around the compute and memory limitations of current systems when used for AI training.

Scientific ML on Disaggregated Cognitive Simulation HPC Platforms

On behalf of Brian Van Essen, informatics group leader at LLNL, Vassilieva presented the Livermore group’s federation of the Lassen massively parallel compute cluster with a CS-1 WSE. The goal, she said, is to introduce machine learning steps into traditional simulations in an intimate and iterative way that speeds attainment of accuracy. The team is using inertial confinement fusion at LLNL’s National Ignition Facility as a testbed for the approach, called “cognitive simulation.”

Simulations must simplify natural phenomena to bring the computational burden down to a manageable level. Often, this means that their predictions don’t match experimental findings. Cognitive simulation improves a simulation’s accuracy using machine learning at different levels of a simulation job.

At the “in the loop” level, ML inferences are made at every time step of the computation. The “on the loop” level consists of ML training or inference at every ~1,000 time steps. “Around the loop” training or inference happens with each simulation. Finally, with the addition of experimental data, “outside the loop” transfer learning occurs every ~10,000 simulations. The combination allows frequent training and potentially very-high-frequency inference to accelerate the simulation. The approach leverages vast quantities of data generated by the simulations, couples simulations with experimental results and provides more accurate predictions for complex multi-physics nature of ICF than possible with traditional simulation-only modeling.

Stream-AI-MD: Streaming AI-Driven Adaptive Molecular Simulations for Heterogeneous Computing Platforms

Arvind Ramanathan of ANL and the University of Chicago presented an application of machine learning to another traditional HPC domain, that of molecular simulations.

“The general idea is we want to implement…machine learning training on the fly, as simulations are running,” he said.

Pursued by traditional HPC means, multiscale simulations, for example those of spike-protein dynamics in the SARS-CoV-2 viral particle, can generate hundreds of terabytes of data. The visualization task is huge, he said: “It’s humanly impossible to peek into biologically interesting events.”

Inserting an iterative, ML-driven loop between successive simulations and analytics has proved a promising means of refining model results, predicting folded, unfolded and misfolded states without human supervision. The method has to date improved resolution and accuracy of atomic contacts within the protein structure with a 50X speedup in sampling folded states. Using the approach, the team acquired a 10,000-fold acceleration of sampling effectiveness compared with traditional molecular dynamics simulations running on specialized hardware.

Atomistic Machine Learning Potentials on Neocortex

“We want to model what we call a potential energy surface, and we want to do it at quantum accuracy,” said Keith Phuthi, a PhD student working with Matthew Guttenberg in Venkat Viswanathan’s group at Carnegie Mellon University. Traditional “density functional theory cannot scale to [the] hundreds to thousands of atoms” needed in many problems in physics, chemistry and materials science. PSC’s Neocortex offered a route beyond that limitation, he added.

The empirical potentials method provides a much simpler analytic form that reduces the cost of the computation in terms of steps per atom. But it is much more approximate and often doesn’t capture details in a molecule’s photoelectronic properties that are required for certain applications. Machine learning potentials offer a bridge between the two methods, offering a better balance of accuracy with computational cost. But current GPU systems limit the data that can be used to train a model, with poor scaling to boot.

By computing invariant atomic features on its SuperDome Flex Server and feeding the data into a CS-1 for prediction, Neocortex enabled Phuthi to model the energy potential of each atom in a given compound as a neural network, summing them to obtain the potential for the molecule.

“Our goal with the early programming [was] to get to this target where we work with much bigger datasets and much bigger molecules than are typically trained on,” and to determine how that scaleup affects accuracy,” Phuthi said. The group has to date run batch sizes larger than 32,000 samples, as compared to a memory-driven limit of about 200 on a GPU system. While the team hasn’t yet optimized parameters, the prediction accuracies are similar.

Physics-Informed Neural Networks (PINNs) for Navier-Stokes Equation

Khemraj Shukla, Assistant professor in the CRUNCH group led by George Em Karniadakis of Brown University, described his use of the WSE technology in solving the Navier-Stokes Equations for the motion of viscous fluids using neural networks. 

“Most of these typical systems have very few high-dimensional data points, but very sparse selection,” he said. “In a conventional approach, it requires forward modeling to run many times to do the system identification (described by partial differential equations), whereas by using  PINNs we can solve the forward and inverse problem with few data-points in one shot”. 

To date Shukla has executed his code for lid driven cavity flow at a Reynold’s number (Re) of 100 on Neocortex, taking a total of 150 seconds. This low Re, signifying a fluid with high relative viscosity and a tendency to exhibit laminar flow, represented a modest starting point for the computations. A similar computation on a V100 GPU took about 10 minutes. Future efforts will include larger Re, which represent lower-viscosity fluids with more complex, turbulent flow, and creating an application programming interface for integrating automated differentiation into the computations.

Wafer-Scale Engines for More than AI

A very different application of WSE technology formed the basis of the final presentation in the panel. Instead of AI, Dirk Van Essendelft, PI of the AI/ML Enhanced CFD group at NETL, described using a CS-1 in cooperation with Cerebras for direct physical simulations for phenomena such as astrophysical events.

“The principle of locality is important” for describing the physics of interacting objects, Van Essendelft said. “This principle holds true for almost all physical systems, outside of quantum entanglement.”

The lack of “spooky action at a distance” outside the quantum realm enables simulators to approximate answers by dividing a problem into a grid of discrete cells. Since each cell in the grid only interacts with its immediate neighbors, at one level the computation is simple. The problem arises when a scientist wants to achieve fine resolution, and the cells become numerous.

Ideally, Van Essendelft said, computing hardware would reproduce the 3D grid as a 3D matrix of processors, with each processor holding the description of the cell it represents and interacting with its immediate neighbors. Conventional distributed computing reproduces this ideal very poorly, with a limited number of processors, slow internal communication, non-localized memory and access to neighboring memory taking thousands of cycles.

The 2D grid of the CS-1’s processors offers a better mirror of the model’s physical grid. With each processor directly interacting with its four nearest neighbors, the local memory can hold field values for a column of cells in the 3D grid, offering access to local and neighbor memory in only a single cycle. Van Essendelft’s BiCgStab Solver achieved near-linear scaling in calculating flow parameters in both a 370-cubed-cell and a 600-cubed-cell mesh.

Footnote: The panelists plan to organize a Cerebras technologies user group to foster information exchange on this promising technology. If you are interested, send an email to [email protected].

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!

What We Know about Alice Recoque, Europe’s Second Exascale System

June 24, 2024

Europe officially announced its second exascale system, Alice Recoque, and you can expect to see that name on the Top500 supercomputer list in a few years. Alice Recoque is the new name for a supercomputer with the opera Read more…

Spelunking the HPC and AI GPU Software Stacks

June 21, 2024

As AI continues to reach into every domain of life, the question remains as to what kind of software these tools will run on. The choice in software stacks – or collections of software components that work together to Read more…

HPE and NVIDIA Join Forces and Plan Conquest of Enterprise AI Frontier

June 20, 2024

The HPE Discover 2024 conference is currently in full swing, and the keynote address from Hewlett-Packard Enterprise (HPE) CEO Antonio Neri on Tuesday, June 18, was an unforgettable event. Other than being the first busi Read more…

Slide Shows Samsung May be Developing a RISC-V CPU for In-memory AI Chip

June 19, 2024

Samsung may have unintentionally revealed its intent to develop a RISC-V CPU, which a presentation slide showed may be used in an AI chip. The company plans to release an AI accelerator with heavy in-memory processing, b Read more…

ASC24 Student Cluster Competition: Who Won and Why?

June 18, 2024

As is our tradition, we’re going to take a detailed look back at the recently concluded the ASC24 Student Cluster Competition (Asia Supercomputer Community) to see not only who won the various awards, but to figure out Read more…

Qubits 2024: D-Wave’s Steady March to Quantum Success

June 18, 2024

In his opening keynote at D-Wave’s annual Qubits 2024 user meeting, being held in Boston, yesterday and today, CEO Alan Baratz again made the compelling pitch that D-Wave’s brand of analog quantum computing (quantum Read more…

Spelunking the HPC and AI GPU Software Stacks

June 21, 2024

As AI continues to reach into every domain of life, the question remains as to what kind of software these tools will run on. The choice in software stacks – Read more…

HPE and NVIDIA Join Forces and Plan Conquest of Enterprise AI Frontier

June 20, 2024

The HPE Discover 2024 conference is currently in full swing, and the keynote address from Hewlett-Packard Enterprise (HPE) CEO Antonio Neri on Tuesday, June 18, Read more…

Slide Shows Samsung May be Developing a RISC-V CPU for In-memory AI Chip

June 19, 2024

Samsung may have unintentionally revealed its intent to develop a RISC-V CPU, which a presentation slide showed may be used in an AI chip. The company plans to Read more…

Qubits 2024: D-Wave’s Steady March to Quantum Success

June 18, 2024

In his opening keynote at D-Wave’s annual Qubits 2024 user meeting, being held in Boston, yesterday and today, CEO Alan Baratz again made the compelling pitch Read more…


Argonne’s Rick Stevens on Energy, AI, and a New Kind of Science

June 17, 2024

The world is currently experiencing two of the largest societal upheavals since the beginning of the Industrial Revolution. One is the rapid improvement and imp Read more…

Under The Wire: Nearly HPC News (June 13, 2024)

June 13, 2024

As managing editor of the major global HPC news source, the term "news fire hose" is often mentioned. The analogy is quite correct. In any given week, there are Read more…

Labs Keep Supercomputers Alive for Ten Years as Vendors Pull Support Early

June 12, 2024

Laboratories are running supercomputers for much longer, beyond the typical lifespan, as vendors prematurely deprecate the hardware and stop providing support. Read more…

MLPerf Training 4.0 – Nvidia Still King; Power and LLM Fine Tuning Added

June 12, 2024

There are really two stories packaged in the most recent MLPerf  Training 4.0 results, released today. The first, of course, is the results. Nvidia (currently 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…

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…

Everyone Except Nvidia Forms Ultra Accelerator Link (UALink) Consortium

May 30, 2024

Consider the GPU. An island of SIMD greatness that makes light work of matrix math. Originally designed to rapidly paint dots on a computer monitor, it was then 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…

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…

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…

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…

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


Nvidia Shipped 3.76 Million Data-center GPUs in 2023, According to Study

June 10, 2024

Nvidia had an explosive 2023 in data-center GPU shipments, which totaled roughly 3.76 million units, according to a study conducted by semiconductor analyst fir 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…

Intel’s Next-gen Falcon Shores Coming Out in Late 2025 

April 30, 2024

It's a long wait for customers hanging on for Intel's next-generation GPU, Falcon Shores, which will be released in late 2025.  "Then we have a rich, a very Read more…


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…

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…

AMD Clears Up Messy GPU Roadmap, Upgrades Chips Annually

June 3, 2024

In the world of AI, there's a desperate search for an alternative to Nvidia's GPUs, and AMD is stepping up to the plate. AMD detailed its updated GPU roadmap, w Read more…

Q&A with Nvidia’s Chief of DGX Systems on the DGX-GB200 Rack-scale System

March 27, 2024

Pictures of Nvidia's new flagship mega-server, the DGX GB200, on the GTC show floor got favorable reactions on social media for the sheer amount of computing po 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