Metadata-Version: 2.4
Name: jaxoplanet
Version: 0.0.3
Summary: Probabilistic models for exoplanet inference in JAX
Project-URL: Homepage, https://jax.exoplanet.codes
Project-URL: Source, https://github.com/exoplanet-dev/jaxoplanet
Project-URL: Bug Tracker, https://github.com/exoplanet-dev/jaxoplanet/issues
Author-email: Dan Foreman-Mackey <foreman.mackey@gmail.com>
License: MIT License
License-File: LICENSE
Classifier: Development Status :: 4 - Beta
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Classifier: Programming Language :: Python
Classifier: Programming Language :: Python :: 3
Requires-Python: >=3.10
Requires-Dist: equinox
Requires-Dist: jax
Requires-Dist: jax<=0.4.31
Requires-Dist: jaxlib
Requires-Dist: jpu>=0.0.2
Requires-Dist: pint
Provides-Extra: comparison
Requires-Dist: numpy<1.22; extra == 'comparison'
Requires-Dist: starry; extra == 'comparison'
Requires-Dist: tqdm; extra == 'comparison'
Requires-Dist: xarray<2023.10.0; extra == 'comparison'
Provides-Extra: dev
Requires-Dist: black; extra == 'dev'
Requires-Dist: gmpy2; extra == 'dev'
Requires-Dist: mpmath; extra == 'dev'
Requires-Dist: nox; extra == 'dev'
Requires-Dist: pre-commit; extra == 'dev'
Requires-Dist: pytest; extra == 'dev'
Requires-Dist: ruff; extra == 'dev'
Requires-Dist: sympy; extra == 'dev'
Provides-Extra: docs
Requires-Dist: arviz; extra == 'docs'
Requires-Dist: corner; extra == 'docs'
Requires-Dist: ipywidgets; extra == 'docs'
Requires-Dist: jaxopt; extra == 'docs'
Requires-Dist: matplotlib; extra == 'docs'
Requires-Dist: myst-nb; extra == 'docs'
Requires-Dist: numpyro; extra == 'docs'
Requires-Dist: numpyro-ext; extra == 'docs'
Requires-Dist: sphinx-autoapi<3.2.0; extra == 'docs'
Requires-Dist: sphinx-book-theme; extra == 'docs'
Requires-Dist: sphinx-design; extra == 'docs'
Provides-Extra: test
Requires-Dist: batman-package; extra == 'test'
Requires-Dist: exoplanet-core; extra == 'test'
Requires-Dist: pytest; extra == 'test'
Requires-Dist: pytest-xdist; extra == 'test'
Provides-Extra: test-math
Requires-Dist: gmpy2; extra == 'test-math'
Requires-Dist: mpmath; extra == 'test-math'
Description-Content-Type: text/markdown

# jaxoplanet

_Astronomical time series analysis with JAX_

---

*jaxoplanet* is a
[functional-programming](https://en.wikipedia.org/wiki/Functional_programming)-forward
implementation of many features from the
[exoplanet](https://docs.exoplanet.codes/en/latest/) and
[starry](https://starry.readthedocs.io/en/latest/) packages built on top of
[JAX](https://jax.readthedocs.io/en/latest/).

*jaxoplanet* includes fast and robust implementations of many exoplanet-specific
operations, including solving Kepler's equation, and computing limb-darkened
light curves. Since *jaxoplanet* is built on top of JAX it has first-class
support for hardware acceleration using GPUs and TPUs, and it also integrates
seamlessly with modeling tools like
[NumPyro](https://numpyro.readthedocs.io/en/latest/), and
[Flax](https://flax.readthedocs.io/en/latest/).

**For the most complete documentation, check out the documentation page at
[jax.exoplanet.codes](https://jax.exoplanet.codes).**

## Installation

You'll first need to install JAX following [the instructions in the JAX
docs](https://jax.readthedocs.io/en/latest/#installation). For example, to
install the CPU version of JAX, you can run:

```bash
python -m pip install "jax[cpu]"
```

Then install `jaxoplanet` with:

```bash
python -m pip install jaxoplanet
```

If you run into issues with installing `jaxoplanet`, specifically on ARM (i.e., M series chips) Macs, take a look at [the
installation instructions](https://jax.exoplanet.codes/en/latest/install).

```{admonition} Navigating the docs
:class: tip

💽 After [installing](doc:install) `jaxoplanet`, head over to the [Quickstart](quickstart) page to see some of its features.

🚩 If you're running into some problems with `jaxoplanet`, check out the [common issues](doc:commonissues) page for some general tips and tricks.

⚡ In the [Tutorials](tutorials) section we've added a few tutorials showcasing some common astronomy problems where `jaxoplanet` might come in handy!

🖥️ Take a look [here](doc:api) to see a user-friendly API reference for commonly used objects. The full API reference is also available!

```

## Attribution

While we don't yet have a citation for `jaxoplanet`, please reference the GitHub
repository if you find this code useful in your research. The BibTeX entry for
the repo is:

```
@software{jaxoplanet,
  author       = {Soichiro Hattori and
                  Lionel Garcia and
                  Catriona Murray and
                  Jiayin Dong and
                  Shashank Dholakia and
                  David Degen and
                  Daniel Foreman-Mackey},
  title        = {{exoplanet-dev/jaxoplanet: Astronomical time series analysis with JAX}},
  month        = mar,
  year         = 2024,
  publisher    = {Zenodo},
  version      = {v0.0.2},
  doi          = {10.5281/zenodo.10736936},
  url          = {https://doi.org/10.5281/zenodo.10736936}
}
```

## License
Copyright (c) 2021-2024 Simons Foundation, Inc.

`jaxoplanet` is free software made available under the MIT License. For details
see the `LICENSE` file.
