Le DoE va tester le calcul de l’IA Cerebras dans les simulations d’armes nucléaires

Les laboratoires américains Sandia National Labs ont déclaré cette semaine qu’ils enquêteraient sur l’utilisation des puces accélératrices de la taille d’une plaquette de Cerebras pour déterminer si les armes nucléaires du pays fonctionneraient comme prévu, si jamais une annihilation mondiale était souhaitée.

Avec le soutien des laboratoires nationaux Lawrence Livermore et Los Alamos, le déploiement sera supervisé par l’Administration nationale de la sécurité nucléaire (NNSA) du ministère de l’Énergie, qui est chargée, entre autres, de maintenir la fiabilité et de prolonger la durée de vie de la ville. ogives grâce à l’utilisation de simulations exécutées sur des superordinateurs. Ces simulations rassurent l’agence sur le fait que toute modification de l’arsenal nucléaire des États-Unis – comme le maintien de la viabilité des packages de physique en remplaçant les matériaux ou en modifiant les conceptions – n’aura pas d’impact inacceptable sur le potentiel destructeur.

Étant donné que la plupart d’entre nous ont convenu de ne plus effectuer de tests réels de ces appareils, des simulations avec des données d’expériences sous-critiques sont nécessaires à la place. Et donc, le silicium de Cerebras sera testé pour voir s’il peut aider ici.

“Cette collaboration avec Cerebras Systems a un grand potentiel pour avoir un impact sur les futures applications de mission en permettant l’intelligence artificielle et les technologies d’apprentissage automatique, qui sont un composant émergent de nos charges de travail de simulation de production”, a déclaré Simon Hammond, qui, en tant que responsable de programme fédéral, supervise les systèmes informatiques et les logiciels. au sein de l’équipe Advanced Simulation and Computing (ASC) de la NNSA.

C’est une mention intéressante de l’IA : les puces de Cerebras sont conçues pour accélérer ce type de travail, et il existe un certain intérêt à utiliser des modèles d’apprentissage automatique pour prédire le résultat d’expériences scientifiques, par opposition à l’approche informatique classique de modélisation des interactions physiques. L’utilisation de l’IA pourrait être plus rapide que le calcul pur, bien que la précision puisse être sacrifiée et qu’un hybride des deux approches puisse être préférable.

Les systèmes CS-2 de Cerebras sont dotés d’une grande puce de la taille d’une assiette remplie de 2,6 billions de transistors. La startup soutient que cette puce “waferscale” surdimensionnée permet un traitement beaucoup plus rapide d’énormes ensembles de données car les informations peuvent rester sur le processeur plus longtemps, ou tout le temps, ce qui évite de mélanger les données dans et hors de la mémoire système plus lente.

Le parvenu est l’un des nombreux à explorer l’informatique à l’échelle des tranches pour accélérer des charges de travail AI/ML plus importantes. Tesla, par exemple, a présenté son supercalculateur Dojo à Hot Chips cette année. Pour une analyse complète de l’architecture de calcul waferscale de Cerebras ou de la plate-forme Dojo de Tesla, consultez notre site partenaire La prochaine plateforme.

Parler avec Le registre, Sivasankaran Rajamanickam, un ingénieur impliqué dans le déploiement de la technologie Cerebras chez Sandia, a exprimé son intérêt à examiner comment l’architecture gère les modèles clairsemés et les flux de données sur puce. “L’échelle du matériel rend vraiment excitant de voir ce que nous pouvons en faire”, a-t-il déclaré.

Cerebras n’est que la dernière startup d’IA à déployer son matériel dans le cadre du programme ASC. Le Dept of Energy explore régulièrement des plates-formes de calcul hétérogènes utilisant une variété de CPU, GPU, NIC et autres accélérateurs pour améliorer la vitesse et la résolution de ces simulations. À ce jour, l’agence a utilisé des systèmes d’Intel, AMD, Graphcore, Fujitsu, Marvell, IBM et Nvidia pour n’en nommer que quelques-uns.

“Nous prévoyons que les technologies développées dans le cadre du programme seront testées sur les systèmes prototypes d’architecture avancée du programme Advanced Simulation and Computing et affecteront éventuellement la production de plates-formes technologiques avancées et de base utilisées par les trois laboratoires”, Robert Hoekstra, directeur principal du groupe informatique à grande échelle chez Sandia, a déclaré dans un communiqué.

On nous dit que les résultats de ces essais éclaireront les investissements futurs du DoE. ®