jaxoplanet

使用JAX进行天文时间序列分析

jaxoplanet 是在 函数式编程 方向实现的，基于 exoplanet 和 starry 包的功能，这些包构建在 JAX 之上。

jaxoplanet 包含了许多针对系外行星的特定操作的快速和稳健的实现，包括解凯普勒方程和计算边缘昏暗的光谱。由于 jaxoplanet 是基于 JAX 构建的，因此它具有一流的硬件加速支持，包括GPU和TPU，并且它可以与建模工具如 NumPyro 和 Flax 无缝集成。

对于最完整的文档，请查看 jax.exoplanet.codes 上的文档页面。

安装

您首先需要根据 JAX文档中的说明 安装JAX。例如，要安装JAX的CPU版本，您可以运行

python -m pip install "jax[cpu]"

然后使用以下命令安装 jaxoplanet

python -m pip install jaxoplanet

如果您在安装 jaxoplanet 时遇到问题，请参阅 安装说明。

快速入门

归属

尽管我们还没有 jaxoplanet 的引用，但如果您在研究中发现此代码很有用，请参考GitHub存储库。该存储库的BibTeX条目是

@software{jaxoplanet,
  author = {{Foreman-Mackey}, D. and {Garcia}, L.~J. and {Hattori}, S.
            and {Dong}, J. and {Murray}, C.},
  title = {{jaxoplanet}: Astronomical time series analysis with {JAX}},
  url = {http://github.com/exoplanet-dev/jaxoplanet},
  version = {0.0.1},
  year = {2024},
}

许可

版权所有（c）2021-2024西蒙斯基金会。

jaxoplanet 是在MIT许可证下提供的免费软件。有关详细信息，请参阅LICENSE文件。