우주

#JAX

13건의 기사

관측·천문현상

Sequince: 순차 몬테카를로(SMC) 알고리즘을 위한 모듈형 파이썬 툴킷 공개

연구진이 순차 몬테카를로(SMC) 알고리즘을 재사용 가능한 구성요소로 조립할 수 있는 모듈형 파이썬 툴킷 'sequince'를 공개했다. 가중치 부여, 리샘플링, 변이, 변환, 제어 단계를 모듈로 제공하며 템퍼드 SMC, 지속 샘플링, SMC 기반 네스티드 샘플링용 사전 구축 샘플러도 포함한다. 핵심 구성요소가 파이썬 Array API 표준을 따라 NumPy, JAX, PyTorch, CuPy를 지원해 CPU와 GPU 모두에서 분석이 가능하고, 난수 제어와 체크포인팅으로 재현 가능한 연구 워크플로를 지원한다. 이번 논문은 sequince v0.1.0과 그 설계를 SMC 방법론 연구 및 활용을 위한 재사용 가능한 인프라로 설명한다.

arXiv astro-ph.IM (관측기기)

천문·천체물리

FEADME 공개: GPU 가속 타원 강착원반 모델링 프레임워크로 DPE·TDE 237개 분석

연구진이 상대론적 타원 강착원반 모델로 광폭 발머 선 방출을 분석하는 GPU 가속 파이썬 프레임워크 FEADME(Fast Elliptical Accretion Disk Modeling Engine)를 공개했다. JAX와 NumPyro 기반 미분가능 순방향 모델링과 베이즈 추론을 적용해 이중 피크 방출체(DPE) 237개와 원반형 Hα 방출을 보이는 조석 교란 현상(TDE) 5건에 대해 스펙트럼별 3개 모델 계열을 적합했고, 사후분포 품질 필터링 후 원반을 가진 활동성 은하핵(AGN) 표본 165개와 TDE 27개 유효 에포크를 남겼다. 공통 타원 원반 모델에서 AGN은 가우시안 혼합 형태 5개 구간으로 요약되는 연속적이고 넓은 원반 기하·운동학 분포를 보였으며, TDE 에포크는 상대적으로 낮은 이심률과 더 넓은 비원반 가우시안 성분을 나타냈다. 같은 AGN 스펙트럼에 대한 기존 원형 원반+나선팔 적합 결과와 비교하면 경사각·반경·방출률 기울기·국소 선폭 분포가 크게 달라, 이들 추정값이 선 프로파일 비대칭을 어떻게 다루는지에 강하게 의존함을 보여준다. 대부분의 AGN과 TDE는 원반에 추가 광폭 성분이 포함된 모델을 선호해, 원반 방출이 보다 등방적이거나 항성풍 기원 가스와 흔히 공존함을 시사한다.

arXiv astro-ph.HE (고에너지)

천문·천체물리

비등방 속도분포 천체 평형모델 생성 파이썬 패키지 'PIANISTpy' 공개

암흑물질 헤일로와 은하는 병합·비등방 질량강착을 거치며 비등방 속도분포를 갖게 되지만, 기존 수치 시뮬레이션 연구는 대체로 등방 속도분포를 가정해 왔다. 유한한 공간 범위에 대해 자기일관적인 절단 경계조건을 갖춘 비등방 초기조건을 만들 도구가 없었기 때문으로, 연구진은 위상공간에서 절단을 수행하는 해석적 틀과 함께 JAX 기반 GPU 가속 오픈소스 패키지 PIANISTpy(Particle Initial conditions with ANISoTropy)를 처음으로 제시했다. 이 패키지는 구형·비회전 초기조건 생성, 임의의 이중멱법칙 밀도 프로파일과 해석적 비등방 프로파일 라이브러리를 지원하며, 3차원 속도 프로파일과 시선방향 적분 관측량을 미분 가능 형태로 출력해 경사도 기반 몬테카를로 추론에 활용할 수 있다. 연구진은 기존 도구인 galpy, AGAMA 대비 수치 solver 정확도가 수 자릿수 높고 입자 초기조건의 역학적 안정성도 우수하다고 밝혔다.

arXiv astro-ph.GA (은하)

천문·천체물리

LISA 다변량 잡음 추정용 종단간 파이프라인 'lisa-noise' 공개

레이저 간섭계 우주 안테나(LISA)의 관측 데이터는 글로벌 피트 기반 분석 전에 필터링·다운샘플링·분할·윈도잉 등의 후처리를 거쳐야 하며, 각 단계가 잡음 과정을 변형시켜 잡음 공분산 행렬의 보정이 필요하다. 연구진은 JAX 기반 잡음 추정 파이프라인 lisa-noise를 제시해, 후처리 효과를 반영한 잡음 스펙트럼 추정값을 내놓고 마이컬슨 채널의 3×3 다변량 공분산 행렬 전체를 피팅했다. 팔 길이가 서로 다르고 일정하지 않은 조건을 가정했으며, 비정상(non-stationary) 효과인 전달함수 영점의 이동은 각 분할 구간 내 여러 에포크에 걸쳐 전달함수를 평균해 반영했다. 30일분의 시뮬레이션 데이터에서 7일 구간의 가능도 평가는 현재 노트북에서 3밀리초가 소요됐고, 좁은 사전분포에서 출발한 경사 기반 피팅은 1초도 안 되는 시간에 주입된 스펙트럼을 복원했다. 시뮬레이션 참값을 중심으로 log10 기준 ±10 범위의 넓은 사전분포에서 다중 시작점 탐색을 수행한 결과도 데이터와 일치하는 잡음 추정값을 얻었고, 베이즈 기법으로 검증됐다.

arXiv astro-ph.IM (관측기기)

천문·천체물리

jax-SNANA 공개: GPU로 초당 10만 개 초신성 시뮬레이션하는 JAX 기반 초신성 분석 프레임워크

초신성 분석 프레임워크 SNANA를 JAX로 재구현한 jax-SNANA가 공개됐다. NVIDIA A100 GPU에서 단정밀도로 구동할 때 LSST 관측 초신성을 초당 10만 개 이상 시뮬레이션하고, JIT 컴파일 후 초당 5,000개의 광도곡선을 피팅해 기존 도구보다 두 자릿수(약 100배) 빠른 성능을 낸다. SALT·BayeSN 분광에너지분포(SED) 모델과 redback-jax의 물리 모델을 지원하며, 측광·분광 적색편이 모두를 이용한 광도곡선 피팅이 가능하다. 연구팀은 기존 SNANA와 비교해 다양한 작업에서 두 결과가 잘 일치함을 확인했으며, 시뮬레이션 기반 추론(SBI) 학습 루프 안에서 즉석 시뮬레이션을 생성할 수 있도록 설계했다. 코드는 깃허브에 공개되어 pip로 설치할 수 있다.

arXiv astro-ph.IM (관측기기)

NASA and industry leaders speak to NASA Social participants about the agency's Transiting Exoplanet Survey Satellite (TESS) in the Press Site auditorium at Kennedy Space Center in Florida. Speaking to
천문·천체물리

ELISA: X선·감마선 스펙트럼 베이지안 분석 오픈소스 툴킷 공개

X선·감마선 천문학의 스펙트럼 분석 전 과정(데이터 로딩, 모델 구성, 추론, 모델 검증·비교)을 지원하는 오픈소스 파이썬 툴킷 ELISA(Efficient Library for Spectral Analysis)가 arXiv를 통해 소개됐다. ELISA는 JAX로 가속한 전방 모델(forward model)과 통합 피팅 인터페이스를 결합해 빠른 최적화와 베이지안 샘플링을 모두 지원하며, 수렴 점검과 예측 평가를 위한 진단 기능을 기본 제공한다. 연구진은 Insight-HXMT의 게성운(Crab) 관측 데이터를 이용한 스펙트럼 피팅 예제에서 XSPEC의 기본 MCMC 구현에 비해 샘플링 효율이 크게 개선됐다고 밝혔다. ELISA는 깃허브(github.com/wcxve/elisa)와 PyPI(astro-elisa)에서 받을 수 있다.

arXiv astro-ph.HE (고에너지)

천문·천체물리

hyprfine: GPU에서 암흑시대부터 재이온화 시기까지 21cm 신호를 시뮬레이션하는 코드 공개

JAX와 파이썬으로 작성된 hyprfine은 적색편이 z=1100~6 구간의 전천 평균 21cm 신호를 GPU 환경에서 해석적으로 계산하는 시뮬레이션 코드다. ΛCDM 우주론 변수와 초기 별·은하의 천체물리에 따라 우주 시간에 걸친 21cm 신호의 평균 온도를 모델링하며, 연구진은 이 코드가 21cm 신호에 대한 최초의 해석적 GPU 네이티브 시뮬레이션이라고 밝혔다. 1초 미만에 실행되고 GPU 전체에 효율적으로 병렬화되며, 암흑시대(z ≥ 35) 구간에서는 미분이 가능하다.

arXiv astro-ph.IM (관측기기)

천문·천체물리

대규모 상미분방정식(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 (관측기기)

A central star distorts spacetime with orbiting planets
천문·천체물리

중력파 파형 대리모델을 JAX로 가속…GPU에서 약 100배 속도 향상

연구진은 머신러닝 기반 중력파 파형 대리모델 mlgw와 mlgw-bns를 JAX 생태계에서 미분 가능하고 하드웨어 가속이 가능한 형태로 구현했다. SEOBNRv5HM 근사 모델로 새로 학습한 대리모델을 쌍성 블랙홀 사례에 적용한 벤치마크에서 CPU 기준으로 기존 mlgw 구현 대비 한 자릿수 이상(10배 이상)의 파형 계산 속도 향상을 보였고, GPU 가속과 대규모 배치 벡터화를 활용하면 약 두 자릿수(100배) 수준에 도달했다. 또한 오픈소스 패키지 GWgpu-jax와 BlackJax-NS 기반 네스티드 샘플링 파이프라인으로 GW150914와 GW170817 이벤트의 모수 추정을 각각 약 12분, 17분 만에 단일 GPU에서 수행했으며, 결과는 LIGO-Virgo-KAGRA 협력단이 보고한 사후분포와 일치했다.

arXiv astro-ph.IM (관측기기)

천문·천체물리

GPU에서 ODE 앙상블을 풀고 미분하는 천체물리 라이브러리 'GRADSOLVE' 공개

해밀토니안 몬테카를로 같은 경사 기반 방법이 천문학 모델 피팅에 쓰이면서, 수천~수백만 개 파라미터 조합에 대해 작은 상미분방정식(ODE) 계를 반복해 푸는 비용이 분석 전체의 부담이 되고 있다. 연구진은 앙상블의 각 구성원을 개별 GPU 스레드에서 각자의 스텝 크기로 적분하고 같은 패스에서 파라미터에 대한 미분값까지 반환하는 JAX 라이브러리 GRADSOLVE를 천문학계에 소개했다. 은하 중력 퍼텐셜 내 별의 궤도, 암흑에너지 우주론의 팽창사, 쌍성 블랙홀의 스핀 세차운동 등 세 가지 사례를 같은 정확도 기준으로 비교한 결과, 100만 개 궤적에서 CPU 128코어 전체를 쓴 직렬 CPU 코드보다 8.5~1500배 빨랐고, 같은 GPU에서 JAX 기반 최신 ODE 라이브러리 DIFFRAX보다 11~15배 빨랐다. 100만 회 적분이 GPU 1대에서 수 초 이내에 끝나 이 규모의 경사 기반 분석이 일상적으로 가능해졌으며, 코드는 GitHub에 공개됐다.

arXiv astro-ph.GA (은하)

천문·천체물리

이상치에 강한 행렬 분해 기법 RHMF 공개… PCA 일반화로 가이아 DR3 특이 항성 탐색

연구진은 주성분분석(PCA)을 일반화한 RHMF(Robust Heteroskedastic Matrix Factorization)를 제안했다. 이 기법은 이상치에 강건하고 특성별 불확실성과 결측 데이터를 처리하며, 특성별·천체별 이상을 자동으로 표시한다. 반복 재가중 알고리즘으로 스튜던트-t 우도를 암묵적으로 최대화하며, 연구진은 JAX 기반 구현체 Robusta-HMF를 함께 공개했다. 가이아 DR3(Gaia DR3)의 RVS 스펙트럼에 적용해 색-등급 공간에서 주변 천체와 다른 주계열성을 찾아냈고, Be형 별을 포함한 알려진 쌍성과 칼슘 II 삼중선에서 미약한 방출선을 보이는 M형 왜성 등 육안으로는 식별하기 어려운 사례를 제시했다.

arXiv astro-ph.IM (관측기기)

천문·천체물리

쌍성 블랙홀 초기데이터의 미분 가능한 파라메트릭 모델 제안: 등각평탄 보웬-요크 펑처 (arXiv 논문)

연구진은 쌍성 블랙홀(binary black hole) 초기데이터를 위한 미분 가능한 파라메트릭 모델을 제시했다. 희소 격자 스펙트럴 보간으로 타원형 solver의 초기 추정값을 제공하고, 소수의 뉴턴(Newton) 반복으로 보정해 이산화된 구속 조건 잔차를 ‖R‖∞≤10⁻¹⁰ 수준까지 낮춘다. JAX로 구현해 물리 파라미터에 대한 해석적 미분값을 제공하며, 등각평탄 보웬-요크(Bowen-York) 펑처를 이용해 4차원 정렬 스핀 모델과 8차원 일반 스핀 모델을 구성했다. 결과는 TwoPunctures와 비교 검증됐고, ADM 질량·각운동량 타기팅과 이심률 감소에 적용됐다.

arXiv astro-ph.IM (관측기기)

Breathtaking view of a starry sky with a majestic nebula in deep space.
천문·천체물리

JAX 기반 펄서 타이밍 통합 패키지 'JUG' 공개… PINT보다 50배 이상 빠르다

연구진이 JAX 기반의 독립형 펄서 타이밍 분석 패키지 JUG(JAX-based Unified pulsar timinG)를 공개했다. JUG는 데이터 처리와 클록 보정부터 타이밍 모델 및 피팅까지 펄서 타이밍 파이프라인 전 과정을 다른 타이밍 소프트웨어에 의존하지 않고 자체 구현했으며, GPU 사용이 가능하고 파이썬 API 또는 대화형 GUI로 조작할 수 있다. 사용자는 복잡한 확률적 잡음을 포함한 타이밍 모델 피팅, 연속 중력파 같은 결정론적 신호 모델링을 수행할 수 있어 빈도주의 타이밍과 베이지안 잡음 분석을 연결한다. 연구진은 JUG가 PINT보다 50배 이상 빠르고 Tempo2와 비슷한 속도를 내며, 수백만 개의 도달 시각(arrival time)을 처리하면서 PINT와 피코초 수준으로 일치한다고 밝혔다.

arXiv astro-ph.HE (고에너지)