Tag: JAX

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.

Differentiable Fluid Solver

Google Research's differentiable JAX-based fluid solver enabling end-to-end gradients for embedding neural networks into CFD time stepping.

PDE Solving

DeepXDE v1.15.0 released with L-LAAF activation function support for JAX backend and Apple MPS GPU acceleration for PyTorch backend, improving deep learning-based PDE solving performance.