Cover Image

JAX-LaB: A High-Performance, Differentiable, Lattice Boltzmann Library for Modeling Multiphase Fluid Dynamics in Geosciences and Engineering

Fuente: arXiv

Saved in:

Bibliographic Details
Main Authors:	Pradhan, Piyush, Gentine, Pierre, Kelly, Shaina
Format:	Preprint
Published:	2025
Subjects:	Computational Physics Distributed, Parallel, and Cluster Computing Fluid Dynamics
Online Access:	Acceder al recurso
Tags:	Add Tag No Tags, Be the first to tag this record!

Similar Items

In Situ Data Summaries for Flexible Feature Analysis in Large-Scale Multiphase Flow Simulations
by: Dutta, Soumya, et al.
Published: (2022)

GPU-Accelerated Simulations of Problems with Moving Boundaries and Fluid-Structure Interaction at Extreme Scales
by: Kumar, Sushrut, et al.
Published: (2026)

A GPU-ready pseudo-spectral method for direct numerical simulations of multiphase turbulence
by: Roccon, Alessio
Published: (2024)

MFC 5.0: An exascale many-physics flow solver
by: Wilfong, Benjamin, et al.
Published: (2025)

waLBerla-wind: a lattice-Boltzmann-based high-performance flow solver for wind energy applications
by: Schottenhamml, Helen, et al.
Published: (2023)

Massive parallelization and performance enhancement of an immersed boundary method based unsteady flow solver
by: Sundar, Rahul, et al.
Published: (2024)

A Comparison of Low and high-Order Methods for the Simulation of Supersonic Jet Flows
by: Abreu, D. F., et al.
Published: (2025)

SPUMA: a minimally invasive approach to the GPU porting of OPENFOAM
by: Bnà, Simone, et al.
Published: (2025)

Sampling-based Distributed Training with Message Passing Neural Network
by: Kakka, Priyesh, et al.
Published: (2024)

Proof-of-concept: Using ChatGPT to Translate and Modernize an Earth System Model from Fortran to Python/JAX
by: Zhou, Anthony, et al.
Published: (2024)

An efficient multi-GPU implementation for the Discontinuous Galerkin ocean model SLIM
by: Court, Miguel De Le, et al.
Published: (2026)

Oceananigans.jl: A Julia library that achieves breakthrough resolution, memory and energy efficiency in global ocean simulations
by: Silvestri, Simone, et al.
Published: (2023)

miniLB: A Performance Portability Study of Lattice-Boltzmann Simulations
by: Crisci, Luigi, et al.
Published: (2024)

Experience and Analysis of Scalable High-Fidelity Computational Fluid Dynamics on Modular Supercomputing Architectures
by: Karp, Martin, et al.
Published: (2024)

SCALE-TRACK: Asynchronous Euler-Lagrange particle tracking on heterogeneous computing architecture
by: Schmalfuß, Silvio, et al.
Published: (2026)

DrJAX: Scalable and Differentiable MapReduce Primitives in JAX
by: Rush, Keith, et al.
Published: (2024)

Energy efficiency: a Lattice Boltzmann study
by: Turisini, Matteo, et al.
Published: (2024)

Near Real-time Adaptive Isotropic and Anisotropic Image-to-mesh Conversion for Numerical Simulations Involving Cerebral Aneurysms
by: Garner, Kevin, et al.
Published: (2024)

On The Performance of Prefix-Sum Parallel Kalman Filters and Smoothers on GPUs
by: Särkkä, Simo, et al.
Published: (2025)

JAXMg: A multi-GPU linear solver in JAX
by: Wiersema, Roeland
Published: (2026)

Portable High-Performance Kernel Generation for a Computational Fluid Dynamics Code with DaCe
by: Andersson, Måns I., et al.
Published: (2025)

Architecture Specific Generation of Large Scale Lattice Boltzmann Methods for Sparse Complex Geometries
by: Suffa, Philipp, et al.
Published: (2024)

Orbax: Distributed Checkpointing with JAX
by: Gaffney, Colin, et al.
Published: (2026)

Design Principles of Dynamic Resource Management for High-Performance Parallel Programming Models
by: Huber, Dominik, et al.
Published: (2024)

Ladon: High-Performance Multi-BFT Consensus via Dynamic Global Ordering (Extended Version)
by: Lyu, Hanzheng, et al.
Published: (2024)

Parallel Data Object Creation: Towards Scalable Metadata Management in High-Performance I/O Library
by: Li, Youjia, et al.
Published: (2025)

Fully Lattice-Linear Algorithms
by: Gupta, Arya Tanmay, et al.
Published: (2022)

Exploring the Emerging Technologies within the Blockchain Landscape
by: Tareq, Mohammad Ali, et al.
Published: (2024)

Exploration on Highly Dynamic Graphs
by: Saxena, Ashish, et al.
Published: (2026)

A Simple Lower Bound for Set Agreement in Dynamic Networks
by: Fraigniaud, Pierre, et al.
Published: (2025)

JORA: JAX Tensor-Parallel LoRA Library for Retrieval Augmented Fine-Tuning
by: Tahir, Anique, et al.
Published: (2024)

Use Cases for High Performance Research Desktops
by: Henschel, Robert, et al.
Published: (2024)

Automatic Metadata Capture and Processing for High-Performance Workflows
by: Shpilker, Polina, et al.
Published: (2025)

Raptr: Prefix Consensus for Robust High-Performance BFT
by: Tonkikh, Andrei, et al.
Published: (2025)

A Blockchain-Enabled Framework for Storage and Retrieval of Social Data
by: Parab, Aishwarya, et al.
Published: (2025)

Highly Dynamic and Fully Distributed Data Structures
by: Augustine, John, et al.
Published: (2024)

HP2C-DT: High-Precision High-Performance Computer-enabled Digital Twin
by: Iraola, E., et al.
Published: (2025)

The Fused Kernel Library: A C++ API to Develop Highly-Efficient GPU Libraries
by: Amoros, Oscar, et al.
Published: (2025)

Bounding Distance Between Outputs in Distributed Lattice Agreement
by: Rasheed, Abdullah, et al.
Published: (2025)

LAPIS: A Performance Portable, High Productivity Compiler Framework
by: Kelley, Brian, et al.
Published: (2025)