Taylor-Green Vortex
Simulations of the Taylor-Green vortex on GPUs.
Simulations of the Taylor-Green vortex, a canonical benchmark problem in fluid dynamics. Below is an $N=64$ run on a molab GPU. Jupyter code for the same run on a colab T4 is hosted on GitHub with a link at the bottom of the page. Both were implemented in JAX. Further below on this page is a run on an NVIDIA H200 using nvmath and CuPy with $N=512$.
Open interactive demo ↗ JAX pseudospectral solver · runs live on a molab GPU
Opens in a new tab. Adjust the sliders, then click Run simulation. The GPU kernel takes a few seconds to warm up on first run.
The full simulation notebook
The notebook below is the $N = 512$ production run on an NVIDIA H200 using nvmath and CuPy with the full derivation, solver, diagnostics and with all output as it was produced. It is a rendered copy. Use the molab demo above to actually run a simulation at lower $N$. GitHub code links are present at the bottom of the page.
Run it yourself
Interactive demo (molab) Source on GitHub
The GitHub link points at the repo containing (1) the JAX implementation, which runs at $N = 64$ on a free Colab T4, and (2) the $N = 512$ run presented above, using an H200.