Efficient Differentiable n-d PDE Solvers in JAX.
-
Updated
Mar 2, 2026 - Jupyter Notebook
Efficient Differentiable n-d PDE Solvers in JAX.
Incompressible Navier-Stokes solver
[Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (≥46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled Training; Rollout Metrics)
[IROS 2024] [ICML 2024 Workshop Differentiable Almost Everything] MonoForce: Learnable Image-conditioned Physics Engine
Electromagnetics simulation library for moving point charges built on JAX
A library for soft differentiable relaxations of common JAX functions.
A library for soft differentiable relaxations of common PyTorch functions.
[NeurIPS 2024] NeuMA: Neural Material Adaptor for Visual Grounding of Intrinsic Dynamics
A simple JAX-powered simulation library for numerical experiments of fuzzy dark matter, stars, gas + more!
AeroJAX: A differentiable, structure-preserving framework for real-time flow simulation, control, and inverse design. Architected for neural operator integration and latent-space acceleration. Built with JAX.
Differentiable Flexible Mechanical Metamaterials
This repo contains the differentiable physics simulation module in "PPR: Physically Plausible Reconstruction from Monocular Videos". ICCV 23.
Universal Partial Differential Equations Simulator
Generic framework for building differentiable models.
Simple differentiable approximate ocean models built with JAX.
Repository for MHPI differentiable hydrological models.
Training methodologies for autoregressive neural operators/emulators in JAX.
A differentiable astrophysics simulator in JAX
Differentiable form-finding with variational inference for architectural structures.
Controlling fluids in a reduced-dimensional simulation.
Add a description, image, and links to the differentiable-physics topic page so that developers can more easily learn about it.
To associate your repository with the differentiable-physics topic, visit your repo's landing page and select "manage topics."