Waferscale, möt atomär skala: Uncle Sam för att testa Cerebras-chips i kärnvapen-sims

Amerikas Sandia National Labs sa denna vecka att de kommer att undersöka användningen av Cerebras acceleratorchips i skivstorlek för att fastställa att landets kärnvapen kommer att fungera som avsett, om global förintelse någonsin skulle önskas.

Med stöd från Lawrence Livermore och Los Alamos nationella laboratorier kommer utplaceringen att övervakas av Dept of Energy's National Nuclear Security Administration (NNSA), som bland annat har till uppgift att upprätthålla tillförlitligheten och förlänga livslängden för stadsutplåning stridsspetsar genom användning av simuleringar som körs på superdatorer. Dessa simuleringar försäkrar byrån att eventuella förändringar av USA:s kärnvapenarsenal – som att hålla fysikpaketen livskraftiga genom att ersätta material eller justera designen – inte kommer att påverka den destruktiva potentialen på ett oacceptabelt sätt.

Eftersom de flesta av oss har gått med på att inte längre utföra verkliga tester av dessa enheter, simuleringar med data från subkritiska experiment behövs istället. Och så kommer Cerebras kisel att testköras för att se om det kan hjälpa här.

"Detta samarbete med Cerebras Systems har stor potential att påverka framtida uppdragstillämpningar genom att möjliggöra artificiell intelligens och maskininlärningsteknik, som är en framväxande komponent i våra produktionssimuleringsarbetsbelastningar," sade Simon Hammond, som som federal programchef övervakar beräkningssystem och mjukvara vid NNSA:s Advanced Simulation and Computing (ASC) team.

Det är ett intressant omnämnande av AI: Cerebras chips är designade för att påskynda den här typen av arbete, och det finns en hel del intresse i att använda maskininlärningsmodeller för att förutse resultatet av vetenskapliga experiment i motsats till det klassiska beräkningssättet att modellera fysiska interaktioner. Att använda AI kan vara snabbare än ren beräkning, även om noggrannhet kan offras, och en hybrid av de två metoderna kan vara bäst.

Cerebras CS-2-system har ett stort chip i storleken middagstallrik packat med 2.6 biljoner transistorer. Uppstarten hävdar att detta superstora "waferscale"-chip tillåter mycket snabbare bearbetning av enorma datamängder eftersom informationen kan finnas kvar på processorn längre, eller hela tiden, vilket undviker att blanda data in och ut ur långsammare systemminne.

Uppkomlingen är en av flera som utforskar waferscale computing för att accelerera större AI/ML-arbetsbelastningar. Tesla, till exempel, demonstrerade sin Dojo superdator på Hot Chips i år. För en fullständig uppdelning av Cerebras waferscale beräkning arkitektur eller Teslas Dojo-plattform, kolla in vår systersida Nästa plattform.

Att tala med Registret, Sivasankaran Rajamanickam, en ingenjör involverad i utbyggnaden av Cerebras-teknologin på Sandia, uttryckte intresse för att undersöka hur arkitekturen hanterar glesa modeller och dataflöden på chipet. "Omfattningen av hårdvaran gör det riktigt spännande att se vad vi kan göra med den," sa han.

Cerebras är bara den senaste AI-starten som distribuerar sin hårdvara under ASC-programmet. Dept of Energy utforskar rutinmässigt heterogena beräkningsplattformar med hjälp av en mängd processorer, GPU:er, NIC:er och andra acceleratorer för att förbättra hastigheten och upplösningen av dessa simuleringar. Hittills har byrån anställd system från Intel, AMD, Graphcore, Fujitsu, Marvell, IBM och Nvidia för att nämna några.

"Vi förväntar oss att teknik som utvecklats som en del av programmet kommer att testas på Advanced Simulation and Computing-programmets avancerade arkitekturprototypsystem och så småningom kommer att påverka produktionen av avancerade och råvaruteknologiplattformar som används av de tre labben," Robert Hoekstra, senior manager för extrem skala datorgrupp på Sandia, sade i ett uttalande.

Vi får veta att resultaten av dessa försök kommer att informera framtida investeringar av DoE. ®