ExoJAX

Differentiable spectral modelling of exoplanets/brown dwarfs/M dwarfs using JAX! Read the docs 🐕. 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, blackjax, bayeux
• Optimizers: JAXopt, 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 for the first step!

Real Examples (external)

• ⭐ exojaxample_WASP39b : An example of HMC-NUTS for actual hot Saturn (JWST/ERS, NIRSPEC/G395H)
• ⭐ exojaxample_jupiter : An example of HMC-NUTS for actual Jupiter reflection spectrum

References

• Paper I: Kawahara, Kawashima, Masuda, Crossfield, Pannier, van den Bekerom, ApJS 258, 31 (2022)
• Paper II: in prep

License

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