XLB: Accelerated Lattice Boltzmann (XLB) for Physics-based ML
☆445Mar 9, 2026Updated last week
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☆538Mar 4, 2026Updated 2 weeks ago
- Computational Fluid Dynamics based on PyTorch and the Lattice Boltzmann Method☆281Updated this week
- A differentiable PDE solving framework for machine learning☆1,837Mar 13, 2026Updated last week
- Computational Fluid Dynamics in JAX☆927Feb 24, 2026Updated 3 weeks ago
- Differentiable Finite Element Method with JAX☆606Feb 8, 2026Updated last month
- JAX-DIPS is a differentiable interfacial PDE solver.☆50Sep 14, 2024Updated last year
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆526Mar 10, 2026Updated last week
- Simple GPU rendering of scientific data with Pytorch, Jax, CuPy, and Warp backends.☆29Nov 20, 2023Updated 2 years ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆133Feb 20, 2026Updated last month
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆519Mar 2, 2026Updated 2 weeks ago
- Curated list of some open source codes employing lattice Boltzmann methods☆248Jul 10, 2024Updated last year
- JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework☆79Oct 29, 2025Updated 4 months ago
- JAX Arrays for human consumption☆110Oct 23, 2025Updated 4 months ago
- Additive manufacturing simulation with JAX.☆341Jul 11, 2025Updated 8 months ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆554Mar 9, 2026Updated last week
- A 3D sparse LBM solver implemented using Taichi☆372Apr 7, 2025Updated 11 months ago
- A Python framework for accelerated simulation, data generation and spatial computing.☆6,326Updated this week
- Efficient Differentiable n-d PDE Solvers in JAX.☆165Mar 2, 2026Updated 2 weeks ago
- 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…☆42Jun 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,930Updated this week
- TCLB - Templated MPI+CUDA/CPU Lattice Boltzmann code☆201Mar 12, 2026Updated last week
- Minimal JAX implementation of k-nearest neighbors using a k-d tree.☆55Jul 15, 2025Updated 8 months ago
- Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/☆1,933Feb 23, 2026Updated 3 weeks ago
- S2FFT: Differentiable and accelerated spherical transforms☆220Updated this week
- Matrix-free linear algebra in JAX.☆161Dec 2, 2025Updated 3 months ago
- Multi-GPU Framework for Voxel Grid Computations☆64Updated this week
- Neural Emulator Architectures in JAX.☆24Feb 20, 2026Updated last month
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆317Updated this week
- ☆49Feb 13, 2026Updated last month
- Python 2D Navier-Stokes solver☆28Aug 4, 2025Updated 7 months ago
- Solve sparse linear systems in JAX using the KLU algorithm☆38Feb 19, 2026Updated last month
- Implementation of Forward Laplacian algorithm in JAX☆94Nov 25, 2025Updated 3 months ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆121Mar 5, 2026Updated 2 weeks ago
- A curated list of repositories related to fluid dynamics.☆348Feb 19, 2026Updated last month
- JAX-accelerated nuclear equation of state inference and TOV solvers☆10Mar 13, 2026Updated last week
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆304Updated this week
- Solver-in-the-Loop: Learning from Differentiable Physics to Interact with Iterative PDE-Solvers☆176Oct 19, 2022Updated 3 years ago
- Stencil computations in JAX☆71Nov 15, 2025Updated 4 months ago