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

위대한 융합의 탐색과 언어의 장벽

지구상에 별을 만드는 것을 상상해 보십시오. 그것이 바로 태양의 에너지를 만드는 과정이자, 깨끗하고 무한한 에너지를 약속하는 핵융합의 꿈입니다. 이를 실현하기 위해 과학자들은 토카막(tokamak)이라 불리는 자기장 감옥 안에 전하를 띤 입자들이 소용돌이치는 뜨거운 플라즈마를 가둡니다. 하지만 이 플라즈마는 매우 혼란스럽습니다. 마치 수십억 마리의 작고 초조한 벌들이 서로 끊임없이 부딪히며 방향을 바꾸는 경로를 예측하려는 것과 같습니다. 이 플라즈마를 어떻게 안정적으로 유지하고 융합할 수 있을 만큼 뜨겁게 만들지 이해하기 위해, 과학자들은 거대한 컴퓨터 시뮬레이션을 사용합니다. 이 시뮬레이션은 입자가 어떻게 움직이고 전자기장이 어떻게 뒤틀리고 회전하는지를 추적하는 가상 실험실 역할을 합니다.

이 작업에서 가장 신뢰받는 도구 중 하나는 ORB5라고 불리는 코드입니다. ORB5를 수십 년 동안 동일한 완벽한 융합 레시피를 요리해 온 숙련된 노련한 셰프로 생각하십시오. 하지만 이 셰프는 오직 한 가지 언어, 즉 1950년대에 발명된 프로그래밍 언어인 포트란(Fortran)만을 구사합니다. 포트란은 수학 계산에는 훌륭하지만, 마치 빈티지 자동차처럼 되어가고 있습니다. 즉, 수리하기 어렵고, 업그레이드하기 힘들며, 최신 그래픽 카드를 탑재한 최첨단 슈퍼컴퓨터와 같은 현대적인 도로에서 운전하기가 어렵습니다. 컴퓨팅의 세계가 새로운 하드웨어로 이동함에 따라, 이 "빈티지 자동차"를 효율적으로 계속 운행하는 것은 점점 더 힘든 과제가 되고 있습니다. 문제는 레시피가 좋은가 아닌가가 아닙니다. 레시피는 훌륭합니다. 문제는 어떻게 그 레시피의 맛을 바꾸지 않으면서 오늘날의 가장 빠른 엔진에서 실행될 수 있도록 현대적인 언어로 번역하느냐 하는 것입니다.

번역 프로젝트: ORB5X

이 논문은 그 유명한 융합 코드의 대담하고 새로운 버전인 ORB5X를 소개합니다. 연구팀은 새로운 레시피를 발명하거나 플라즈마의 물리학을 바꾸려 하지 않았습니다. 대신, 그들은 숙련된 번역가 역할을 했습니다. 그들은 원래의 포트란 코드를 가져와서 "에이전틱 AI(Agentic AI, 코드를 작성하고 디버깅하는 스마트한 컴퓨터 조수)"의 도움을 받아 전체 프로그램을 현대적인 프로그래밍 언어인 **C++**로 다시 작성했습니다.

이것이 왜 중요할까요? 여러분에게 아주 훌륭한 집 설계도가 있지만, 그것이 빛이 바래가고 특정 유형의 제도판에만 맞게 그려져 있다고 상상해 보십시오. ORB5X는 그 똑같은 설계도를 어떤 컴퓨터, 어떤 태블릿, 심지어 우주에 있는 슈퍼컴퓨터에서도 작동하는 디지털 태블릿 위에 다시 그리는 것과 같습니다. 목표는 원래 코드의 "수학적 영혼"을 보존하면서도 이를 "성능 이식 가능(performance-portable)"하게 만드는 것이었습니다. 이는 새 코드가 표준 노트북, 강력한 데스크톱, 또는 수천 개의 그래픽 카드를 가진 거대한 슈퍼컴퓨터 어디에서나 재작성 없이도 잘 실행될 수 있음을 의미합니다.

어떻게 수행했는가: AI 부조종사

번역은 단순히 복사해서 붙여넣는 작업이 아니라, 세심하고 단계적인 재구성 과정이었습니다. 연구진은 인간 전문가와 AI 에이전트가 함께 협력하는 워크플로우를 사용했습니다. AI는 지치지 않는 부조종사 역할을 했습니다. AI는 오래된 포트란 코드의 한 덩어리를 살펴보고, 이를 현대적인 C++로 어떻게 작성할지 제안하며, 수학적 오류가 발생할 수 있는 잠재적인 함정을 표시했습니다. 그러면 인간 과학자들이 모든 제안을 검토하여 새로운 코드가 기존 코드와 정확히 동일하게 작동하는지 확인했습니다.

그들은 두 버전을 마치 이중 언어 사전처럼 나란히 두었습니다. 만약 기존 코드에 fields.F90이라는 파일이 있다면, 새로운 코드에는 fields.cpp가 있었습니다. 이를 통해 모든 계산을 하나하나 확인할 수 있었습니다. 또한 그들은 **코코스(Kokos)**라는 현대적인 툴킷을 사용했는데, 이는 일종의 범용 어댑터 역할을 합니다. 코코스는 프로그래머가 각기 다른 기기를 위해 별도의 명령을 작성할 필요 없이, 코드가 다양한 유형의 컴퓨터 프로세서(예: Intel CPU 또는 NVIDIA GPU)와 자동으로 통신할 수 있게 해줍니다.

결과: 기계적 정확도

연구팀은 다양한 플라즈마 난류를 시뮬레이션하는 세 가지 유명한 "스트레스 테스트"를 통해 ORB5X를 시험대에 올렸습니다:

1. ITG (이온 온도 구배): 플라즈마를 통해 열이 어떻게 이동하는지 시뮬레이션합니다.
2. ITPA: 빠른 입자와 자기파가 관련된 복잡한 테스트입니다.
3. 처핑(Chirping): 에너지 입자와 상호작용할 때 파동의 주파수가 어떻게 변하는지 관찰합니다.

결과는 놀라웠습니다. 새로운 C++ 코드의 출력값을 원래의 포트란 코드와 비교했을 때, 수치가 **기계적 정확도(machine accuracy)**로 일치했습니다. 이는 두 코드 사이의 차이가 컴퓨터의 수학적 정밀도에 의해 제한될 정도로 극히 미미하여 사실상 제로에 가깝다는 것을 의미합니다. 개인용 노트북에서 실행하든 LUMI나 CINECA와 같은 거대한 슈퍼컴퓨터에서 실행하든, 물리학은 동일하게 유지되었습니다.

이것이 왜 중요한 일인가

이 논문은 ORB5X가 성공적임을 확인시켜 주지만, 동시에 현실적인 기대치도 설정합니다. 새 코드가 모든 기기에서 반드시 기존 포트란 버전보다 더 빠른 것은 아닙니다. 실제로 기존 코드는 수십 년간 미세 조정되었기 때문에 일부 경우에는 여로 여전히 약간 더 빠릅니다. 그러나 진정한 승리는 **이식성(portability)**에 있습니다. 기존의 포트란 코드는 새로운 하드웨어에서 실행하기 어려운 "레거시(legacy)" 시스템이 되어가고 있습니다. 현대적인 도구로 구축된 ORB5X는 개인용 노트북(Apple M4 칩이 탑재된 Mac 포함)과 세계에서 가장 진보된 슈퍼컴퓨터 모두에서 쉽게 컴파일되고 실행될 수 있습니다.

저자들은 이것이 버전 1.0이라는 점을 주의 깊게 언급합니다. 이것은 완성된 제품이 아니라 견고한 토대입니다. 그들은 향후 컴퓨터 메모리와 프로세서 사이의 데이터 이동 시간을 줄임으로써 코드를 더욱 빠르게 만들 계획이며, 궁극적으로는 더 복잡한 형태의 자기장 감옥(예: 스텔라레이터)을 시뮬레이션하는 데 사용하기를 희망합니다. 하지만 현재로서는, 수십 년 된 매우 복잡한 과학 코드를 AI의 도움을 받아 현대적인 언어로 번역하면서도 물리학을 완벽하게 유지할 수 있음을 성공적으로 증명해 냈습니다. 이는 과거와 미래의 융합 연구를 잇는 가교이며, 깨끗한 에너지를 향한 여정이 세상에서 가장 빠른 엔진 위에서도 계속 달려 나갈 수 있도록 보장하는 것입니다.