DiFVM puts an unstructured-mesh finite-volume solver on the GPU: one card beats 32-core OpenFOAM and still yields end-to-end gradients
DiFVM rewrites the complete finite-volume/PISO solution procedure on unstructured polyhedral meshes as vectorized scatter and gather operations over a static graph, giving GPU-native execution and end-to-end automatic differentiation in JAX/XLA. Existing differentiable CFD platforms mostly rely on Cartesian or structured grids and do not directly handle body-conforming complex geometries. On a single NVIDIA L40 GPU, DiFVM is faster than 32-core parallel OpenFOAM in every tested case, the largest with about 106 control volumes; differentiability is demonstrated by recovering a cavity lid velocity and inferring six Windkessel parameters simultaneously.