JAX-LaB: A High-Performance, Differentiable, Lattice Boltzmann Library for Modeling Multiphase Fluid Dynamics in Geosciences and Engineering. - Piyush Pradhan; Pierre Gentine; Shaina Kelly (CoRR, 2025)