发表机构
Imperial College London; I-X Centre for AI in Science, Translation and Innovation Hub (I-HUB), Imperial College London（帝国理工学院; 帝国理工学院 I-X 科学翻译与创新中心（I-HUB））
机构由 AI 辅助整理，请以论文原文为准。AI 中文总结
hyprfine是首个基于JAX和Python的解析性GPU原生模拟，用于快速计算从黑暗时代到再电离纪元的21-cm信号，并支持微分。
AI 中文摘要
hyprfine是一个解析模拟，使用JAX和Python编写，具备原生GPU能力，用于模拟从$z=1100$到$z=6$的天空平均21-cm信号。它模拟了21-cm信号随宇宙时间的平均温度，作为$\Lambda$-CDM宇宙学参数以及第一批恒星和星系天体物理学的函数。据我们所知，该代码是第一个解析性的GPU原生21-cm信号模拟。它能在几分之一秒内运行，在GPU上高效并行化，并且可通过黑暗时代（$z \geq 35$）进行微分。
英文摘要
hyprfine is an analytic simulation of the sky-averaged 21-cm signal from $z=1100 - 6$ written using JAX and Python for native GPU capabilities. It models the average temperature of the 21-cm signal over cosmic time as a function of the $Λ$-CDM cosmology parameters and the astrophysics of the first stars and galaxies. As far as we are aware, the code is the first analytic GPU native simulation of the 21-cm signal. It runs in a fraction of a second, parallelises efficiently across a GPU and is differentiable through the Dark Ages ($z \geq 35$). The code is publicly available at https://www.github.com/htjb/hyprfine.
CommentsSubmitted to JOSS. Comments welcome!
