Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/HajimeKawahara/exojax

🐈 Automatic differentiable spectrum modeling of exoplanets/brown dwarfs using JAX, compatible with NumPyro and JAXopt
https://github.com/HajimeKawahara/exojax

Last synced: about 2 months ago
JSON representation

🐈 Automatic differentiable spectrum modeling of exoplanets/brown dwarfs using JAX, compatible with NumPyro and JAXopt

Host: GitHub
URL: https://github.com/HajimeKawahara/exojax
Owner: HajimeKawahara
License: mit
Created: 2020-12-12T03:46:49.000Z (about 4 years ago)
Default Branch: master
Last Pushed: 2024-07-20T07:55:24.000Z (6 months ago)
Last Synced: 2024-07-22T11:16:20.599Z (6 months ago)
Language: Jupyter Notebook
Homepage: http://secondearths.sakura.ne.jp/exojax/
Size: 187 MB
Stars: 41
Watchers: 4
Forks: 14
Open Issues: 25
Metadata Files:
- Readme: README.md
- Contributing: CONTRIBUTING.md
- License: LICENSE

Awesome Lists containing this project

awesome-jax - exojax - Automatic differentiable spectrum modeling of exoplanets/brown dwarfs compatible to JAX. <img src="https://img.shields.io/github/stars/HajimeKawahara/exojax?style=social" align="center"> (Libraries / New Libraries)

README

        # ExoJAX

 [![License](https://img.shields.io/github/license/HajimeKawahara/exojax)](https://github.com/HajimeKawahara/exojax/blob/develop/LICENSE)

 [![Docs](https://img.shields.io/badge/docs-exojax-brightgreen)](http://secondearths.sakura.ne.jp/exojax/)

 [![arxiv](https://img.shields.io/badge/arxiv-2105.14782-blue)](http://arxiv.org/abs/2105.14782)

 [![paper](https://img.shields.io/badge/paper-ApJS_258_31_(2022)-orange)](https://iopscience.iop.org/article/10.3847/1538-4365/ac3b4d) 

 

Differentiable spectral modelling of exoplanets/brown dwarfs/M dwarfs using JAX!

Read [the docs](http://secondearths.sakura.ne.jp/exojax/develop) 🐕. 

In short, ExoJAX allows you to do gradient based optimizations, HMC-NUTS, and SVI using the latest database.

ExoJAX is at least compatible with

- PPLs: [NumPyro](https://github.com/pyro-ppl/numpyro), [blackjax](https://github.com/blackjax-devs/blackjax), [bayeux](https://github.com/jax-ml/bayeux) 

- Optimizers: [JAXopt](https://github.com/google/jaxopt), [optax](https://github.com/google-deepmind/optax)



ExoJAX Classes

- Databases: *db (mdb: molecular, adb: atomic, cdb:continuum, pdb: particulates)

- Opacity Calculators: opa  (Voigt profile, CIA, Mie, Rayleigh scattering etc)

- Atmospheric Radiative Transfer: art (emission w, w/o scattering, refelction, transmission)

- Spectral Operator: sop (planet rotation, instrumental boradening)

- Atompsheric Microphysics: amp (clouds etc)

## Get Started 

See [this page](http://secondearths.sakura.ne.jp/exojax/develop/tutorials/get_started.html) for the first step!

## Real Examples (external)

- :star: [exojaxample_WASP39b](https://github.com/sh-tada/exojaxample_WASP39b) : An example of HMC-NUTS for actual hot Saturn (JWST/ERS, NIRSPEC/G395H)

- :star: [exojaxample_jupiter](https://github.com/HajimeKawahara/exojaxample_jupiter) : An example of HMC-NUTS for actual Jupiter reflection spectrum

## References

[![paper](https://img.shields.io/badge/paper_I-ApJS_258_31_(2022)-orange)](https://iopscience.iop.org/article/10.3847/1538-4365/ac3b4d) 

- Paper I: Kawahara, Kawashima, Masuda, Crossfield, Pannier, van den Bekerom, [ApJS 258, 31 (2022)](https://iopscience.iop.org/article/10.3847/1538-4365/ac3b4d)

- Paper II: in prep

## License

🐈 Copyright 2020-2024 ExoJAX contributors. ExoJAX is publicly available under the MIT license.