천문·천체물리
대규모 상미분방정식(ODE) 앙상블을 GPU로 가속하는 라이브러리 'modax' 공개
연구진은 서로 독립적인 저차원 상미분방정식(ODE) 앙상블을 대량으로 푸는 데 최적화된 GPU 가속 파이썬 라이브러리 'modax'를 공개했다. JAX 생태계와 호환되지만 순수 JAX 방식 대신 저수준 스레드 기반 프로그래밍 방식으로 작성돼, 궤적 발산이 큰 앙상블에 더 적합하다는 것이 연구진의 설명이다. 로렌츠(Lorenz)·로버트슨(Robertson)·반데르폴(van der Pol) 격자 시스템 벤치마크에서 명시적 Tsit5 솔버는 diffrax 대비 로렌츠 시스템에서 30배, 암시적 Rodas5P 솔버는 로버트슨 시스템에서 700배 이상의 속도 향상을 기록했다. 연구진은 원시 원소 존재비를 이용한 베이지안 모수 추정, 21cm 전역 신호의 몬테카를로 불확실성 정량화, 원시 파워 스펙트럼의 비결합 푸리에 모드 계산 등 우주론 분야 3개 사례에 modax를 적용해 200차원 미만 ODE로 기술되는 물리 현상 모델링에서 효율이 크게 개선됨을 보였다.
arXiv astro-ph.IM (관측기기)