JAX-ALFA Logo

Getting Started

  • Introduction
    • Features
    • About the Author
  • Installation
    • Requirements
    • Getting the Code
    • Installing JAX
    • Installing Other Dependencies
    • Verifying the Installation
    • Running a Simulation
    • Selecting CPU or GPU
    • Running on a SLURM Cluster
  • License

User Guidelines

  • Best Practices and Pitfalls
    • 1. Surface Flux as Lower Boundary Condition for Stable Boundary Layers
    • 2. First Half-Model Level Height for High-Resolution Simulations
    • 3. Filter-to-Grid Ratio (FGR) for LASDD-SM in Stable Boundary Layers
    • 4. Inapplicability of MOST for Free Convective Boundary Layers
  • Config.py Reference
    • 1. Platform
    • 2. Domain
    • 3. Time Integration
    • 4. Surface Boundary Conditions
    • 5. Forcing
    • 6. Subgrid-Scale (SGS) Model
    • 7. Rayleigh Damping Layer
    • 8. Statistics and Output
    • 9. Large-Scale Advection Forcing
    • 10. Moisture
    • 11. Pressure Solver

Tutorial

  • Stable BL: GABLS1
    • 1. Case Overview
    • 2. Prerequisites
    • 3. Directory Layout
    • 4. Step 1 - Inspect Config.py
    • 5. Step 2 — Create Initial Conditions
    • 6. Step 3 — Create the Surface Boundary-Condition File
    • 7. Step 4 — Launch the Solver
    • 8. Step 5 - Load and Inspect the Output
    • 9. Mean Wind and Temperature Profiles
    • 10. Momentum and Heat Fluxes
    • 11. Temporal Evolution of the Friction Velocity
    • 12. Velocity Variances
    • 13. Next Steps

Performance Benchmarks

  • NVIDIA A100 (80 GB)
    • Hardware
    • Case Summary
    • Setup
    • Headline Metrics
    • Completion Matrix
    • Compact Results Table
    • Resolution Scaling
    • Performance Ratio Summary
    • Cross-Case Diagnostic
  • Hardware Platform Comparison
    • Hardware
    • Selected Configurations
    • Setup
    • Results Table
    • Time per Iteration by Platform
    • Speedup Relative to A100 (80 GB)

Case Studies

  • Convective BL
    • Description
    • Resolution Sensitivity
    • SGS Model Sensitivity
    • SGS Coefficients
    • Precision Sensitivity
    • Horizontal Cross-Sections
    • Vertical Cross-Sections
    • Fourier Spectra
  • Neutral BL
    • Description
    • Resolution Sensitivity
    • SGS Model Sensitivity
    • SGS Coefficients
    • Precision Sensitivity
    • Horizontal Cross-Sections
    • Vertical Cross-Sections
    • Fourier Spectra
  • Stable BL: GABLS1
    • Description
    • Resolution Sensitivity
    • SGS Model Sensitivity
    • SGS Coefficients
    • Precision Sensitivity
    • Horizontal Cross-Sections
    • Vertical Cross-Sections
    • Fourier Spectra
  • Stable BL: GABLS3
    • Case Description
    • Time–Height Diagnostics
    • Horizontal Cross-Sections
    • Vertical Cross-Sections
  • Diurnal Cycle: Wangara
    • Description
    • Time–Height Diagnostics
    • Horizontal Cross-Sections
    • Vertical Cross-Sections

Model Structure

  • Overview
    • Initialization Module (INI)
    • Main Program (MAIN)
    • Navier-Stokes Equations (NSE)
    • Scalar Equations (SCL)
    • Subgrid-Scale (SGS) Modeling
    • Surface Layer (SFC) Modeling
    • Core Operations (OPS)
    • Miscellaneous Items (MSC)

Additional Resources

  • API Reference
    • Configuration Loader
    • Derived Variables
    • Import All Modules
    • Initialization of Variables
    • Preprocessing of Variables
    • Main Program
    • All Terms
    • Advection Terms
    • Buoyancy Terms
    • Pressure Terms
    • SGS Terms
    • Time Advancement
    • Scalar All Terms
    • Scalar Advection Terms
    • Scalar SGS Terms
    • Scalar Time Advancement
    • Strain Rates Computations
    • SGS Stress Computations: Smagorinsky
    • SGS Stress Computations: Wong-Lilly
    • Scalar SGS Flux Computations: Smagorinsky
    • Scalar SGS Flux Computations: Wong-Lilly
    • Static SGS Modeling: Main Code
    • Dynamic SGS Modeling: Main Code
    • SGS Model: LASDD-SM (Momentum)
    • SGS Model: LASDD-WL (Momentum)
    • SGS Model: LASDD-SM (Scalar)
    • SGS Model: LASDD-WL (Scalar)
    • Surface Flux Parameterization
    • FFT Computations
    • Derivative Computations
    • Dealiasing Computations
    • Filtering Computations
    • Compute Statistics
    • Miscellaneous Terms
JAX-ALFA
  • Search

© Copyright 2025–2026, Sukanta Basu.

Built with Sphinx using a theme provided by Read the Docs.