Supercomputer-scale in a single wafer-scale system
Scientific computing powers the forefront of research and discovery in fundamental physics and life sciences, leading to important advances in industry, security, and insights that improve and enrich our daily lives.
The workloads that drive these applications often require massive sparse computation with high bandwidth memory access and communication. Even supercomputers are challenged by this work.
Using the CS-2, researchers can achieve orders of magnitude acceleration for traditional scientific computing workloads, AI surrogate workloads, or HPC+AI cognitive simulations that leverage both.
Computational fluid dynamics
Computational fluid dynamics (CFD) codes are a central component of aerospace and energy research. These workloads require massive sparse computation at extraordinary memory and communication bandwidth. Researchers typically take great lengths to develop state of the art implementations on massive supercomputers. The Cerebras WSE-2 puts all of these resources on a single chip, delivering supercomputer-impossible performance of 100s-10,000s of legacy C/GPU machines for this work.
AI-accelerated modeling & simulation
Traditional HPC can be accelerated by AI surrogate models that learn physics or by AI models that augment traditional supercomputers to inform the next steps of a physics-based simulation code. As the world’s fastest AI computer, the CS-2 is uniquely well-suited as a specialist accelerator for this type of work in a heterogenous AI+HPC cluster.
Molecular dynamics simulations power a range of applications from drug discovery to materials science. Calculating the time dependent state and interactions across large-scale simulations typically requires hundreds of nodes on a supercomputer. Deep neural networks are increasingly used for the prediction of energies and forces in molecules. With the CS-2, researchers can train physics-informed neural networks and accelerate MD simulations codes on a single machine with the WSE-2’s high bandwidth local on-wafer SRAM and interconnect.
At Argonne National Laboratory (ANL), Cerebras Systems is working with research staff in the Computing, Environment, and Life Sciences (CELS) directorate to accelerate groundbreaking scientific and technical innovation in biology and life sciences.
Associate Laboratory Director of Computing, Environment and Life Sciences @ Argonne National Laboratory