JAX-ALFA

A JAX-based large-eddy simulation framework for atmospheric boundary-layer flows.

JAX-ALFA — Atmospheric LES For All — is a JAX-based large-eddy simulation (LES) framework for atmospheric boundary-layer flows, developed in my group.

Features

  • Spectral methods for horizontal derivatives; finite differences in the vertical
  • FFT-based pressure solver
  • Subgrid-scale closures with dynamic (Germano-type) coefficient computation
  • Runs on CPUs and GPUs; single- and double-precision
  • Written in Python / JAX; GPL-3.0-or-later
  • Under active development

Citing

If you use JAX-ALFA in research, teaching, or derivative software, please cite the GitHub repository and documentation. A formal citation / DOI will be added when available.