google-deepmind/torax

해결하는 문제

TORAX는 핵융합 에너지 연구에 사용되는 장치인 토카막 내 플라즈마의 핵심 수송을 모델링하기 위해 설계된 시뮬레이터입니다. 복잡한 물리 계산을 위한 매우 효율적이고 미분 가능한 환경을 제공함으로써 과학자들이 빠른 순방향 모델링, 펄스 설계, 궤적 최적화 및 컨트롤러 설계를 수행할 수 있도록 돕습니다.

작동 원리

Python과 JAX를 사용하여 구축된 TORAX는 자동 미분을 활용하여 비선형 편미분 방정식 (PDE)을 빠르고 정확하게 해결합니다. 미분 가능하기 때문에 시뮬레이터는 수동 유도 없이도 결과가 다양한 입력 매개변수에 얼마나 민감한지 계산할 수 있습니다. 이를 통해 머신러닝 대리 모델을 통합하고, 궤적을 최적화하며, 반경험적 모델의 매개변수를 식별하는 것이 쉬워집니다. 열 수송, 전류 확산, 불순물 복사와 같은 다양한 물리적 과정을 Newton-Raphson 및 최적화 기반 접근 방식을 포함한 여러 솔버 옵션을 사용하여 처리합니다.

대상 사용자

TORAX는 빠르고 확장 가능하며 현대적인 시뮬레이션 도구가 필요한 핵융합 연구원 및 엔지니어를 위한 것입니다. 특히 펄스 설계나 컨트롤러 개발 작업 중 물리 워크플로우에 머신러닝 모델을 통합하거나 경사 기반 최적화를 수행하려는 사용자에게 유용합니다.

주요 특징

• 미분 가능한 아키텍처를 통해 경사 기반 최적화 및 민감도 분석 가능.
• 난류 수송을 위한 QLKNN 모델을 포함한 머신러닝 대리 모델에 대한 내장 지원.
• 유연한 기하학적 지원으로 다양한 평형 파일 형식 사용 가능.
• JAX 컴파일을 통한 고성능 실행 및 실행 시간 단축을 위한 영구 캐싱.
• RAPTOR와 같은 기존 토카막 시뮬레이터와의 검증 완료.