Argonne Talks AI Accelerators for Covid Research

By Oliver Peckham

April 28, 2022

As the pandemic swept across the world, virtually every research supercomputer lit up to support Covid-19 investigations. But even as the world transformed, the fairly stable status quo of simulation-based scientific computing was itself beginning to more rapidly change with the burgeoning field of AI and AI-specific accelerators. The Argonne Leadership Computing Facility (ALCF) at Argonne National Laboratory is, perhaps, the exemplar of these dual shifts, hosting both a wide range of Covid-19 research and the ALCF AI Testbed, equipped with a cornucopia of novel AI accelerators. At the AI Systems Summit this week, Venkatram Vishwanath—data science team lead at ALCF—spoke to the intersection of advanced AI acceleration and Covid-19 research at the lab.

“We recently deployed the ThetaGPU and Polaris systems, which are HPE-Nvidia systems where we’re using A100 GPUs, and we are on our path to deploy Aurora, which is an Intel-HPE system and will have the Intel GPUs,” Vishwanath said. “As we march along these [systems], what we’ve also noticed is the growth of AI accelerators and the promise that they have shown for enterprise workloads.”

That promise, he said, made ALCF interested in how the accelerators could be used to tackle the “surge” in scientific machine learning, with AI being used to replace parts of simulations—or even entire simulations. “Our key goal is to really understand: how do you deploy and design infrastructure to facilitate these novel ways of how HPC and AI are being integrated for science?”

Argonne’s AI Testbed

To that end, ALCF began a pathfinding effort that has, so far, been reified through the AI Testbed. The testbed is currently home to five core systems: a CS-2 system from Cerebras; an MK1 system from Graphcore; a Groq system; a Gaudi system from Habana Labs; and a Dataflow system from SambaNova.

The systems in Argonne’s AI Testbed (click to expand).

“As part of the testbed, we have a wide gamut of science applications that we have witnessed to run effectively on these systems,” Vishwanath continued, “that spans from understanding the drug response for cancer, various imaging science experiments, inference operations in tokamak reactor operations, protein-protein folding, among others.”

“As part of Covid-19, we are really interested in understanding how the SARS-CoV-2 spike protein interacts with the host—and this is a really complex, multiscale problem.”

Vishwanath went on to outline Argonne’s traditional computing workflow for these sorts of science problems. “We would run a large ensemble of jobs, which have different configurations,” he said. “We would then write this out to storage, analyze the data and visualize the results—and then try to understand, perhaps there were some configurations that did not produce results that we wanted, and we would stop these, or we would find new configurations that we should have explored and kickstart new simulations.”

“The key goal is to see: can we do this more intelligently?”

Working smarter (and harder)

To that end, ALCF researchers worked in early 2021 to see if they could combine simulations with online learning to steer simulations towards the configurations they needed to explore.

“We are running, in this case, the OpenMM simulations on the ThetaGPU nodes,” Vishwanath said. “We have this coupled with the [Cerebras] CS-1 system that is on the ALCF AI testbed. We stream the data and the trajectories from Theta-GPU to the CS1. We run online training on the CS-1, where we are trying to use the trajectories to learn this embedding space. We then run our inference on one of the ThetaGPU nodes to identify outliers, which helps us either stop existing simulations or seed new ones as part of this.”

Operating this workflow on a 131,000 system that Vishwanath described as “very critical for Covid-19 research,” he said that “we observe a close to 3.7-fold improvement … where we obtain close to 83,000 samples per second on a CS-1 compared to 2,200 samples per second on our DGX.” (ThetaGPU consists of 24 Nvidia DGX A100 nodes.)

“The other thing to note is that by using the CS-1 accelerator in this case, you are able to completely overlap the training with the compute runs,” he continued. “So the overhead in terms of the overall workflow is mitigated.” This translated to sampling a SARS-CoV-2 spike protein conformation similar to a native folded state (“very key”) in 18 hours of wall clock time. “If we were to run an equilibrium simulation on the A100s … it would have taken us approximately 923 hours,” Vishwanath said. “We were able to explore this space and achieve a 50× improvement over an equilibrium ensemble simulation.”

A comparison of time-to-solution for an eight-hour experiment on an A100-based system versus a Cerebras CS-1 system.

Taking AI acceleration to “new dimensions”

That specific work was completed in 2021—but Vishwanath said that since then, ALCF has “taken it to completely new dimensions.” Specifically, the researchers began looking at how to integrate SARS-CoV-2 research data from cryogenic electron microscopy (cryo-EM) experiments, all-atom molecular dynamics (MD) simulations and fluctuating finite-element analysis (FFEM).

“How do you integrate cryo-EM, FFEA and the all-atom MD simulations to really give you the long timescales that are needed as part of this research?” Vishwanath asked.

The ALCF research team designed a workflow where the cryo-EM data was used as an initial best guess, which was then fed to the FFEA process, which helped identify key variables using hierarchical AI. This was then used to steer all-atom MD simulations, which themselves could be used to help researchers design subsequent cryo-EM experiments.

“In comparison to the previous work, we have, now, three different types of AI that are coupled in,” Vishwanath said. “Some for surrogate modeling, some for intelligent steering of the simulation and for inferencing, among others. And this workflow is also geographically distributed, and it spans components that can run on the CPU, that can run on the GPU … [and] the accelerators that I’ve been describing here.”

This workflow—which enabled research that was nominated for the Gordon Bell Special Prize for High Performance Computing-Based Covid-19 Research in 2021—spanned the Perlmutter system (National Energy Research Scientific Computing Center), the Frontera system (Texas Advanced Computing Center), the Summit system (Oak Ridge Leadership Computing Facility), both Theta and ThetaGPU (ALCF) and the AI Testbed.

Vishwanath said that this geographically and technologically distributed workflow was managed using Balsam. “We were able to map a really diverse set of workflows, such as NAMD 3.0 [which] runs effectively on the GPUs to leverage the resources, FFEA that ran on the CPUs, and the deep learning training and inference jobs that ran on the accelerators.”

The Cerebras CS-2 system.

Once again, he said, Cerebras hardware—specifically, the Wafer-Scale Engine that powers its systems—proved its mettle in the process. (In the interim, Argonne had upgraded its CS-1 system to a CS-2.)

“What we observe is that out of the box, we get about 100× improvement on the Wafer-Scale Engine over a single V100 GPU,” he said. “We expect this to improve as we start optimizing this model on the Wafer-Scale Engine.”

Vishwanath closed by reaffirming how seriously Argonne takes its experiments with novel AI accelerators. “I’ll say that AI accelerators are really promising for scientific machine learning and we’ve observed significant speedups for a wide gamut of science,” he said, adding that Argonne expected such accelerators to play a “key role in future system design.”

To learn more about this research, read additional coverage from HPCwire here.

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!

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

March 18, 2024

Nvidia's latest and fastest GPU, code-named Blackwell, is here and will underpin the company's AI plans this year. The chip offers performance improvements from its predecessors, including the red-hot H100 and A100 GPUs. Read more…

Nvidia Showcases Quantum Cloud, Expanding Quantum Portfolio at GTC24

March 18, 2024

Nvidia’s barrage of quantum news at GTC24 this week includes new products, signature collaborations, and a new Nvidia Quantum Cloud for quantum developers. While Nvidia may not spring to mind when thinking of the quant Read more…

2024 Winter Classic: Meet the HPE Mentors

March 18, 2024

The latest installment of the 2024 Winter Classic Studio Update Show features our interview with the HPE mentor team who introduced our student teams to the joys (and potential sorrows) of the HPL (LINPACK) and accompany Read more…

Houston We Have a Solution: Addressing the HPC and Tech Talent Gap

March 15, 2024

Generations of Houstonian teachers, counselors, and parents have either worked in the aerospace industry or know people who do - the prospect of entering the field was normalized for boys in 1969 when the Apollo 11 missi Read more…

Apple Buys DarwinAI Deepening its AI Push According to Report

March 14, 2024

Apple has purchased Canadian AI startup DarwinAI according to a Bloomberg report today. Apparently the deal was done early this year but still hasn’t been publicly announced according to the report. Apple is preparing Read more…

Survey of Rapid Training Methods for Neural Networks

March 14, 2024

Artificial neural networks are computing systems with interconnected layers that process and learn from data. During training, neural networks utilize optimization algorithms to iteratively refine their parameters until Read more…

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

March 18, 2024

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

Nvidia Showcases Quantum Cloud, Expanding Quantum Portfolio at GTC24

March 18, 2024

Nvidia’s barrage of quantum news at GTC24 this week includes new products, signature collaborations, and a new Nvidia Quantum Cloud for quantum developers. Wh Read more…

Houston We Have a Solution: Addressing the HPC and Tech Talent Gap

March 15, 2024

Generations of Houstonian teachers, counselors, and parents have either worked in the aerospace industry or know people who do - the prospect of entering the fi Read more…

Survey of Rapid Training Methods for Neural Networks

March 14, 2024

Artificial neural networks are computing systems with interconnected layers that process and learn from data. During training, neural networks utilize optimizat Read more…

PASQAL Issues Roadmap to 10,000 Qubits in 2026 and Fault Tolerance in 2028

March 13, 2024

Paris-based PASQAL, a developer of neutral atom-based quantum computers, yesterday issued a roadmap for delivering systems with 10,000 physical qubits in 2026 a Read more…

India Is an AI Powerhouse Waiting to Happen, but Challenges Await

March 12, 2024

The Indian government is pushing full speed ahead to make the country an attractive technology base, especially in the hot fields of AI and semiconductors, but Read more…

Charles Tahan Exits National Quantum Coordination Office

March 12, 2024

(March 1, 2024) My first official day at the White House Office of Science and Technology Policy (OSTP) was June 15, 2020, during the depths of the COVID-19 loc Read more…

AI Bias In the Spotlight On International Women’s Day

March 11, 2024

What impact does AI bias have on women and girls? What can people do to increase female participation in the AI field? These are some of the questions the tech Read more…

Alibaba Shuts Down its Quantum Computing Effort

November 30, 2023

In case you missed it, China’s e-commerce giant Alibaba has shut down its quantum computing research effort. It’s not entirely clear what drove the change. 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…

Analyst Panel Says Take the Quantum Computing Plunge Now…

November 27, 2023

Should you start exploring quantum computing? Yes, said a panel of analysts convened at Tabor Communications HPC and AI on Wall Street conference earlier this y Read more…

DoD Takes a Long View of Quantum Computing

December 19, 2023

Given the large sums tied to expensive weapon systems – think $100-million-plus per F-35 fighter – it’s easy to forget the U.S. Department of Defense is a 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…

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…

Intel’s Server and PC Chip Development Will Blur After 2025

January 15, 2024

Intel's dealing with much more than chip rivals breathing down its neck; it is simultaneously integrating a bevy of new technologies such as chiplets, artificia Read more…

Baidu Exits Quantum, Closely Following Alibaba’s Earlier Move

January 5, 2024

Reuters reported this week that Baidu, China’s giant e-commerce and services provider, is exiting the quantum computing development arena. Reuters reported � Read more…

Leading Solution Providers

Contributors

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…

Training of 1-Trillion Parameter Scientific AI Begins

November 13, 2023

A US national lab has started training a massive AI brain that could ultimately become the must-have computing resource for scientific researchers. Argonne N Read more…

Shutterstock 1179408610

Google Addresses the Mysteries of Its Hypercomputer 

December 28, 2023

When Google launched its Hypercomputer earlier this month (December 2023), the first reaction was, "Say what?" It turns out that the Hypercomputer is Google's t 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…

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…

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…

Google Introduces ‘Hypercomputer’ to Its AI Infrastructure

December 11, 2023

Google ran out of monikers to describe its new AI system released on December 7. Supercomputer perhaps wasn't an apt description, so it settled on Hypercomputer Read more…

China Is All In on a RISC-V Future

January 8, 2024

The state of RISC-V in China was discussed in a recent report released by the Jamestown Foundation, a Washington, D.C.-based think tank. The report, entitled "E Read more…

  • arrow
  • Click Here for More Headlines
  • arrow
HPCwire