** Note:** This assumes CUDA drivers version 12.x. Consult the JAX documentation for other versions. GPU vs CPU: While the code runs significantly faster on GPU, it can also run on CPU-only systems.
This project replicates the classical 2D channel flow simulation using the Lattice Boltzmann Method (LBM) with the D2Q9 lattice model. Implemented in Python and optimized using Numba for performance, ...