XLB: A differentiable massively parallel lattice Boltzmann library in Python

arXiv:2311.16080v3 Announce Type: replace-cross
Abstract: The lattice Boltzmann method (LBM) has emerged as a prominent technique for solving fluid dynamics problems due to its algorithmic potential for computational scalability. We introduce XLB library, a Python-based differentiable LBM library based on the JAX platform. The architecture of XLB is predicated upon ensuring accessibility, extensibility, and computational performance, enabling scaling effectively across CPU, TPU, multi-GPU, and distributed multi-GPU or TPU systems. The library can be readily augmented with novel boundary conditions, collision …

arxiv boltzmann cs.ce cs.lg differentiable lattice library physics.comp-ph python type