XLB: Accelerated Lattice Boltzmann (XLB) based on JAX for Physics-based ML - View it on GitHub