tumaer/JAXFLUIDS
Differentiable Fluid Dynamics Package
This package helps fluid dynamics researchers and engineers simulate complex 3D compressible single-phase and two-phase flows. You provide your flow conditions and geometries, and it outputs detailed flow field data, pressure, velocity, and density distributions, allowing for advanced analysis and optimization. It's designed for those working at the cutting edge of machine learning and computational fluid dynamics.
538 stars. Actively maintained with 9 commits in the last 30 days.
Use this if you need to run high-fidelity simulations of compressible fluid dynamics, including scenarios with multiple fluid phases or moving boundaries, and require automatic differentiation for model optimization on HPC systems.
Not ideal if you are looking for a simple, off-the-shelf CFD tool for basic incompressible flows without the need for advanced differentiation or massive parallel computing.
Stars
538
Forks
99
Language
Python
License
MIT
Category
Last pushed
Mar 04, 2026
Commits (30d)
9
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/tumaer/JAXFLUIDS"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Compare
Related frameworks
explosion/thinc
🔮 A refreshing functional take on deep learning, compatible with your favorite libraries
google-deepmind/optax
Optax is a gradient processing and optimization library for JAX.
patrick-kidger/diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable....
google/grain
Library for reading and processing ML training data.
patrick-kidger/equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/