JAX-Fluids
Differentiable CFD Solver
Fully auto-differentiable CFD solver on JAX (CPU/GPU/TPU) for end-to-end differentiable fluid simulation, coupling seamlessly with neural networks.
GNU General Public License v3.0 JAXDifferentiable CFDDifferentiable Physics