ORB5X v1.0: a performance-portable global electromagnetic gyrokinetic PIC code in C++/Kokkos built using Agentic AI

De Grote Fusie-jacht en de Taalbarrière

Stel je voor dat je een ster probeert te bouwen, hier op aarde. Dat is de droom van kernfusie, het proces dat de zon aandrijft en belooft te voorzien in schone, grenzeloze energie. Om dit te realiseren, vangen wetenschappers superheet plasma in – een kolkende soep van geladen deeltjes – in magnetische kooien die tokamaks worden genoemd. Maar dit plasma is chaotisch; het is alsof je probeert het pad te voorspellen van een miljard piepkleine, hyperactieve bijen die constant tegen elkaar aan botsen en van richting veranderen. Om te begrijpen hoe we dit plasma stabiel en heet genoeg kunnen houden om te fuseren, gebruiken wetenschappers enorme computersimulaties. Deze simulaties fungeren als virtuele laboratoria die bijhouden hoe deeltjes bewegen en hoe elektromagnetische velden draaien en buigen.

Een van de meest vertrouwde instrumenten voor deze taak is een code genaamd ORB5. Denk aan ORB5 als een hooggeschoolde, ouderwetse meesterkok die al decennia lang hetzelfde perfecte fusie-recept bereidt. Echter, deze chef spreekt slechts één taal: Fortran, een programmeertaal die in de jaren 50 werd uitgevonden. Hoewel Fortran geweldig is voor de wiskunde, begint het een beetje op een klassieke auto te lijken: moeilijk te repareren, moeilijk te upgraden en lastig te besturen op moderne wegen (zoals de nieuwste, snelste supercomputers met krachtige grafische kaarten). Naarmate de wereld van computing verschuift naar nieuwe hardware, wordt het een strijd om deze "klassieke auto" efficiënt draaiende te houden. De vraag is niet of het recept goed is – dat is het – maar hoe we dat recept vertalen naar een moderne taal zodat het op de snelste motoren van vandaag kan draaien zonder de smaak van het gerecht te veranderen?

Het Vertalingsproject: ORB5X

Dit artikel introduceert ORB5X, een gedurfde nieuwe versie van die beroemde fusiecode. Het team heeft niet geprobeerd een nieuw recept uit te vinden of de fysica van het plasma te veranderen. In plaats daarvan traden zij op als meestervertalers. Ze namen de originele Fortran-code en schreven het volledige programma in C++, een moderne programmeertaal, met behulp van "Agentic AI" (slimme computerassistenten die code schrijven en debuggen).

Waarom is dit belangrijk? Stel je voor dat je een briljante blauwdruk hebt voor een huis, maar dat deze is getekend op een stuk papier dat vervaagt en alleen past op een specifiek type tekentafel. ORB5X is als het opnieuw tekenen van die exacte blauwdruk op een digitale tablet die werkt op elke computer, elke tablet, of zelfs een supercomputer in de ruimte. Het doel was om de "wiskundige ziel" van de originele code te bewaren terwijl deze "prestatie-draagbaar" (performance-portable) werd gemaakt. Dit betekent dat de nieuwe code net zo goed kan draaien op een standaard laptop, een krachtige desktop of een massieve supercomputer met duizenden grafische kaarten, zonder dat er voor elke machine een volledige herschrijving nodig is.

Hoe ze het deden: De AI Co-piloot

De vertaling was niet zomaar een eenvoudige kopieer-en-plakklus; het was een zorgvuldige, stapsgewijze reconstructie. De onderzoekers gebruikten een workflow waarbij menselijke experts en AI-agenten samenwerkten. De AI fungeerde als een onvermoeibare co-piloot: het keek naar een blok van de oude Fortale-code, suggereerde hoe dit in moderne C++ geschreven kon worden, en gaf waarschuwingen voor potentiële valkuilen waar de wiskunde fout zou kunnen gaan. De menselijke wetenschappers controleerden vervolgens elke suggestie om te garanderen dat de nieuwe code zich exact gedroeg als de oude.

Ze hielden de twee versies naast elkaar, als een tweetalig woordenboek. Als de oude code een bestand had met de naam fields.F90, dan had de nieuwe de naam fields.cpp. Dit stelde hen in staat om elke enkele berekening te controleren. Ze gebruikten een moderne toolkit genaamd Kokkos, die fungeert als een universele adapter. Het laat de code automatisch communiceren met verschillende soorten computerprocessoren (zoals Intel CPU's of NVIDIA GPU's), zonder dat de programmeur aparte instructies voor elk type hoeft te schrijven.

De Resultaten: Machine-nauwkeurigheid

Het team heeft ORB5X getest aan de hand van drie beroemde "stress-tests" die verschillende soorten plasma-turbulentie simuleren:

1. ITG (Ion Temperature Gradient): Het simuleren van hoe warmte door het plasma beweegt.
2. ITPA: Een complexe test die te maken heeft met snel bewegende deeltjes en magnetische golven.
3. Chirping: Het observeren van hoe de frequentie van golven verandert terwijl ze interageren met energetische deeltjes.

De resultaten waren verbluffend. Toen ze de output van de nieuwe C++-code vergeleken met de originele Fortran-code, kwamen de cijfers overeen met machine-nauwkeurigheid. Dit betekent dat het verschil tussen de twee zo klein was dat het in essentie nul was, beperkt door slechts de precisie van de berekeningen van de computer. Of ze de simulatie nu draaiden op een persoonlijke laptop of op een massieve supercomputer zoals LUMI of CINECA, de fysica bleef identiek.

Waarom dit een grote zaak is

Het artikel bevestigt dat ORB5X een succes is, maar stelt ook realistische verwachtingen. De nieuwe code is nog niet noodzakelijkerwijs sneller dan de originele Fortran-versie op elke machine; sterker nog, de originele code is in sommige gevallen nog iets sneller omdat deze decennialang is verfijnd. De echte overwinning is echter draagbaarheid. De originele Fortran-code wordt een "legacy"-systeem dat moeilijk te draaien is op nieuwe hardware. ORB5X, gebouwd met moderne tools, kan gemakkelijk worden gecompileerd en uitgevoerd op persoonlijke laptops (zelfs Macs met Apple M4-chips) en de meest geavanceerde supercomputers ter wereld.

De auteurs merken voorzichtig op dat dit versie 1.0 is. Het is een solide fundament, geen afgewerkt product. Ze zijn van plan het in de toekomst nog sneller te maken door de tijd te verminderen die nodig is om gegevens tussen het geheugen van de computer en de processor te verplaatsen, en ze hopen het uiteindelijk te gebruiken om zelfs complexere vormen van magnetische kooien (zoals stellaratoren) te simuleren. Maar voor nu hebben ze succesvol bewezen dat je een decennia oude, zeer complexe wetenschappelijke code kunt vertalen naar een moderne taal met behulp van AI, en de fysica perfect intact kunt houden. Het is een brug tussen het verleden en de toekomst van fusieonderzoek, die ervoor zorgt dat de zoektocht naar schone energie kan blijven voortbestaan op de snelste motoren die de wereld te bieden heeft.