Computational Fluid Dynamics in JAX
☆927Jan 14, 2026Updated last month
Alternatives and similar repositories for jax-cfd
Users that are interested in jax-cfd are comparing it to the libraries listed below
Sorting:
- Differentiable Fluid Dynamics Package☆532Updated this week
- A differentiable PDE solving framework for machine learning☆1,819Feb 17, 2026Updated last week
- PDEBench: An Extensive Benchmark for Scientific Machine Learning☆1,069Updated this week
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆132Feb 20, 2026Updated last week
- ☆38May 20, 2021Updated 4 years ago
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆1,027Dec 17, 2025Updated 2 months ago
- XLB: Accelerated Lattice Boltzmann (XLB) for Physics-based ML☆442Jan 20, 2026Updated last month
- Differentiable Finite Element Method with JAX☆588Feb 8, 2026Updated 3 weeks ago
- Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/☆1,912Feb 23, 2026Updated last week
- Examples of how to use machine learning algorithms in computational fluid dynamics.☆269Aug 26, 2022Updated 3 years ago
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆516Feb 23, 2026Updated last week
- Learning in infinite dimension with neural operators.☆3,410Updated this week
- Solver-in-the-Loop: Learning from Differentiable Physics to Interact with Iterative PDE-Solvers☆174Oct 19, 2022Updated 3 years ago
- Differentiable, Hardware Accelerated, Molecular Dynamics☆1,381Updated this week
- Lecture material for machine learning applied to computational fluid mechanics☆434Dec 16, 2025Updated 2 months ago
- The versatile ocean simulator, in pure Python, powered by JAX.☆382Feb 23, 2026Updated last week
- A library for scientific machine learning and physics-informed learning☆3,880Dec 12, 2025Updated 2 months ago
- A flexible framework for solving PDEs with modern spectral methods.☆660Updated this week
- All the handwritten notes 📝 and source code files 🖥️ used in my YouTube Videos on Machine Learning & Simulation (https://www.youtube.co…☆1,149Feb 14, 2026Updated 2 weeks ago
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆516Feb 22, 2026Updated last week
- A sequence of Jupyter notebooks featuring the "12 Steps to Navier-Stokes" http://lorenabarba.com/☆3,926Mar 19, 2024Updated last year
- Code for "Learning data-driven discretizations for partial differential equations"☆167Aug 19, 2025Updated 6 months ago
- Curated list for ML in FM☆238Sep 6, 2025Updated 5 months ago
- A framework for fluid flow (Reynolds-averaged Navier Stokes) predictions with deep learning☆336Nov 15, 2022Updated 3 years ago
- ☆510Apr 1, 2025Updated 11 months ago
- PyFR☆542Updated this week
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆107Nov 12, 2023Updated 2 years ago
- ☆290Sep 14, 2024Updated last year
- Differentiable interface to FEniCS for JAX☆58May 30, 2021Updated 4 years ago
- ☆29Jul 8, 2024Updated last year
- This repository is the official implementation of the paper Convolutional Neural Operators for robust and accurate learning of PDEs☆208Nov 24, 2025Updated 3 months ago
- Computational Fluid Dynamics based on PyTorch and the Lattice Boltzmann Method☆280Feb 11, 2026Updated 2 weeks ago
- This repository contains fundamental codes related to CFD that can be included in any graduate level CFD coursework.☆538Aug 23, 2021Updated 4 years ago
- ☆196May 10, 2021Updated 4 years ago
- ☆129Mar 28, 2023Updated 2 years ago
- A curated list of repositories related to fluid dynamics.☆343Feb 19, 2026Updated last week
- Spectral Navier Stokes (and similar) solvers in Python☆343Aug 4, 2023Updated 2 years ago
- PyTorch implementation of the diffusion-based method for CFD data super-resolution proposed in the paper "A Physics-informed Diffusion Mo…☆219May 28, 2023Updated 2 years ago
- Hidden Fluid Mechanics☆354Jan 30, 2023Updated 3 years ago