2026 Project Description

Go back to the list of available projects

JAX-accelerated photochemical modeling

Keywords:

Code development

Exoplanets

Supervisors

Shang-Min Tsai
Find out more about supervisors on ASIAA website

Task Description and Goals

Photochemical models is one of the essential tools for characterizing planetary atmospheres. Solving stiff ODEs is in general the numerical bottleneck. This project aims to implement a JAX-accelerated version of the open-source 1D photochemical model VULCAN to enable faster and more scalable simulations of exoplanet atmospheres.

Depending on the progress, the JAX-version model will be applied to generate datasets for multi-fidelity neural networks.

Required Background

Familiar with Python: Classes, decorators, jit