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

La Gran Cacería de la Fusión y la Barrera del Lenguaje

Imagine intentar construir una estrella aquí mismo en la Tierra. Ese es el sueño de la fusión nuclear, el proceso que alimenta al sol, que promete una energía limpia e ilimitada. Para hacer esto realidad, los científicos atrapan plasma supercaliente —una sopa de partículas cargadas en constante movimiento— en jaulas magnéticas llamadas tokamaks. Pero este plasma es caótico; es como intentar predecir la trayectoria de mil millones de diminutas y hiperactivas abejas que chocan constantemente entre sí y cambian de dirección. Para entender cómo mantener este plasma estable y lo suficientemente caliente como para fusionarse, los científicos utilizan simulaciones computacionales masivas. Estas simulaciones actúan como laboratorios virtuales, rastreando cómo se mueven las partículas y cómo los campos electromagnéticos se retuercen y giran.

Una de las herramientas más confiables para este trabajo es un código llamado ORB5. Piense en ORB5 como un chef maestro altamente capacitado y de la vieja escuela que ha estado cocinando la misma receta de fusión perfecta durante décadas. Sin embargo, este chef habla un solo idioma: Fortran, un lenguaje de programación inventado en la década de 1950. Si bien Fortran es excelente para las matemáticas, se está convirtiendo en algo parecido a un coche clásico: difícil de reparar, difícil de actualizar y difícil de conducir en las carreteras modernas (como las supercomputadoras más nuevas y rápidas con potentes tarjetas gráficas). A medida que el mundo de la computación se desplaza hacia un nuevo hardware, mantener este "coche clásico" funcionando eficientemente se está convirtiendo en una lucha. La pregunta no es si la receta es buena —lo es—, sino ¿cómo traducimos esa receta a un lenguaje moderno para que pueda funcionar en los motores más rápidos de hoy sin cambiar el sabor del plato?

El Proyecto de Traducción: ORB5X

Este artículo presenta ORB5X, una nueva y audaz versión de ese famoso código de fusión. El equipo no intentó inventar una nueva receta ni cambiar la física del plasma. En su lugar, actuaron como maestros traductores. Tomaron el código Fortran original y, con la ayuda de la "IA Agéntica" (asistentes computacionales inteligentes que escriben y depuran código), reescribieron todo el programa en C++, un lenguaje de programación moderno.

¿Por qué es esto importante? Imagine que tiene un plano brillante para una casa, pero está dibujado en un trozo de papel que se está desvaneciendo y que solo encaja en un tipo específico de mesa de dibujo. ORB5X es como redibujar ese mismo plano exacto en una tableta digital que funciona en cualquier computadora, cualquier tableta o incluso en una supercomputadora en el espacio. El objetivo era preservar el "alma matemática" del código original mientras se hacía "portátil en rendimiento". Esto significa que el nuevo código puede ejecutarse tan bien en una computadora portátil estándar, en una potente computadora de escritorio o en una enorme supercomputadora con miles de tarjetas gráficas, sin necesidad de una reescritura completa para cada máquina.

Cómo lo hicieron: El Copiloto de IA

La traducción no fue un simple trabajo de copiar y pegar; fue una reconstrucción cuidadosa y paso a paso. Los investigadores utilizaron un flujo de trabajo donde expertos humanos y agentes de IA trabajaron juntos. La IA actuó como un copiloto incansable: observaba un fragmento del viejo código Fortran, sugería cómo escribirlo en C++ moderno y señalaba posibles trampas donde las matemáticas podrían fallar. Luego, los científicos humanos revisaban cada sugerencia, asegurándose de que el nuevo código se comportara exactamente como el anterior.

Mantuvieron las dos versiones una al lado de la otra, como un diccionario bilingüe. Si el código antiguo tenía un archivo llamado fields.F90, el nuevo tenía fields.cpp. Esto les permitió verificar cada uno de los cálculos. Utilizaron un conjunto de herramientas moderno llamado Kokos, que actúa como un adaptador universal. Permite que el código hable con diferentes tipos de procesadores de computadora (como las CPU de Intel o las GPU de NVIDIA) automáticamente, sin que el programador tenga que escribir instrucciones separadas para cada uno.

Los Resultados: Precisión de Máquina

El equipo puso a prueba ORB5X utilizando tres famosas "pruebas de estrés" que simulan diferentes turbulencias de plasma:

1. ITG (Gradiente de Temperatura Iónica): Simulando cómo se mueve el calor a través del plasma.
2. ITPA: Una prueba compleja que involucra partículas de movimiento rápido y ondas magnéticas.
3. Chirping (Gritar/Variación de frecuencia): Observando cómo cambia la frecuencia de las ondas a medida que interactúan con partículas energéticas.

Los resultados fueron asombrosos. Cuando compararon la salida del nuevo código C++ contra el código Fortran original, los números coincidieron con una precisión de máquina. Esto significa que la diferencia entre los dos era tan minúscula que era esencialmente cero, limitada solo por la precisión de las matemáticas de la computadora. Ya fuera que ejecutaran la simulación en una computadora portátil personal o en una enorme supercomputadora como LUMI o CINECA, la física permanecía idéntica.

Por qué esto es importante

El artículo confirma que ORB5X es un éxito, pero también establece expectativas realistas. El nuevo código no es necesariamente más rápido que la versión Fortran original en cada máquina todavía; de hecho, el código original sigue siendo ligeramente más rápido en algunos casos porque ha sido perfeccionado durante décadas. Sin embargo, la verdadera victoria es la portabilidad. El código Fortan original se está convirtiendo en un sistema "legado" que es difícil de ejecutar en el nuevo hardware. ORB5X, construido con herramientas modernas, puede compilarse y ejecutarse fácilmente en computadoras portátiles personales (incluso en Macs con chips Apple M4) y en las supercomputadoras más avanzadas del mundo.

Los autores advierten cuidadosamente que esta es la versión 1.0. Es una base sólida, no un producto terminado. Planean hacerlo aún más rápido en el futuro reduciendo el tiempo que toma mover datos entre la memoria de la computadora y su procesador, y esperan usarlo eventualmente para simular formas aún más complejas de jaulas magnéticas (como los stellarators). Pero por ahora, han demostrado con éxito que se puede tomar un código científico de décadas de antigüedad y altamente complejo, traducirlo a un lenguaje moderno con la ayuda de la IA y mantener la física perfectamente intacta. Es un puente construido entre el pasado y el futuro de la investigación de la fusión, asegurando que la búsqueda de energía limpia pueda seguir corriendo en los motores más rápidos que el mundo pueda ofrecer.