XLB: Accelerated Lattice Boltzmann (XLB) for Physics-based ML
☆442Jan 20, 2026Updated last month
Alternatives and similar repositories for XLB
Users that are interested in XLB are comparing it to the libraries listed below
Sorting:
- Differentiable Fluid Dynamics Package☆524Jan 29, 2026Updated last month
- Computational Fluid Dynamics based on PyTorch and the Lattice Boltzmann Method☆278Feb 11, 2026Updated 2 weeks ago
- Computational Fluid Dynamics in JAX☆927Jan 14, 2026Updated last month
- A differentiable PDE solving framework for machine learning☆1,819Feb 17, 2026Updated last week
- Differentiable Finite Element Method with JAX☆588Feb 8, 2026Updated 2 weeks ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆132Feb 20, 2026Updated last week
- Simple GPU rendering of scientific data with Pytorch, Jax, CuPy, and Warp backends.☆29Nov 20, 2023Updated 2 years ago
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆516Updated this week
- JAX-DIPS is a differentiable interfacial PDE solver.☆50Sep 14, 2024Updated last year
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆516Updated this week
- Curated list of some open source codes employing lattice Boltzmann methods☆248Jul 10, 2024Updated last year
- JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework☆77Oct 29, 2025Updated 4 months ago
- JAX Arrays for human consumption☆110Oct 23, 2025Updated 4 months ago
- Additive manufacturing simulation with JAX.☆339Jul 11, 2025Updated 7 months ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆544Feb 16, 2026Updated last week
- A 3D sparse LBM solver implemented using Taichi☆370Apr 7, 2025Updated 10 months ago
- Minimal JAX implementation of k-nearest neighbors using a k-d tree.☆54Jul 15, 2025Updated 7 months ago
- A Python framework for accelerated simulation, data generation and spatial computing.☆6,248Updated this week
- OpenFOAM examples for data-driven ML and ROM☆19Jul 25, 2023Updated 2 years ago
- Neural Emulator Architectures in JAX.☆23Feb 20, 2026Updated last week
- Matrix-free linear algebra in JAX.☆160Dec 2, 2025Updated 2 months ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆120Feb 3, 2026Updated 3 weeks ago
- S2FFT: Differentiable and accelerated spherical transforms☆217Updated this week
- A 3D multicomponent multiphase lattice Boltzmann solver with a Multi-Relaxation-Time collision scheme and a sparse storage structure☆30Aug 8, 2023Updated 2 years ago
- This program is a GPU version 3D Lattice Boltzmann Method Computational Fluid Dynamics solver to simulate steady/unsteady 3D single-phase…☆41Jun 5, 2021Updated 4 years ago
- The fastest and most memory efficient lattice Boltzmann CFD software, running on all GPUs and CPUs via OpenCL. Free for non-commercial us…☆4,902Feb 21, 2026Updated last week
- Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/☆1,912Updated this week
- Efficient Differentiable n-d PDE Solvers in JAX.☆145Feb 20, 2026Updated last week
- Official implementation of the AIAA Journal paper "Uncertainty-aware Surrogate Models for Airfoil Flow Simulations with Denoising Diffusi…☆82Nov 4, 2024Updated last year
- Solve sparse linear systems in JAX using the KLU algorithm☆38Feb 19, 2026Updated last week
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆312Feb 9, 2026Updated 2 weeks ago
- A simple JAX solver for the 2D Euler equations for benchmarking☆19Nov 3, 2025Updated 3 months ago
- JAX-accelerated nuclear equation of state inference and TOV solvers☆10Feb 20, 2026Updated last week
- Code repository for "Learned Turbulence Modelling with Differentiable Fluid Solvers"☆42Sep 30, 2022Updated 3 years ago
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆300Jan 9, 2026Updated last month
- ☆45Feb 13, 2026Updated 2 weeks ago
- Implementation of Forward Laplacian algorithm in JAX☆94Nov 25, 2025Updated 3 months ago
- A JAX-based adaptive mesh refinement framework☆28Jan 30, 2026Updated last month
- ☆48May 4, 2024Updated last year