[Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (≥46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled Training; Rollout Metrics)

Incompressible Navier-Stokes solver

[IROS 2024] [ICML 2024 Workshop Differentiable Almost Everything] MonoForce: Learnable Image-conditioned Physics Engine

Electromagnetics simulation library for moving point charges built on JAX

[NeurIPS 2024] NeuMA: Neural Material Adaptor for Visual Grounding of Intrinsic Dynamics

Efficient Differentiable n-d PDE solvers in JAX.

A simple JAX-powered simulation library for numerical experiments of fuzzy dark matter, stars, gas + more!

This repo contains the differentiable physics simulation module in "PPR: Physically Plausible Reconstruction from Monocular Videos". ICCV 23.

Differentiable Flexible Mechanical Metamaterials

Universal Partial Differential Equations Simulator

Simple differentiable approximate ocean models built with JAX.

Generic framework for building differentiable models.

Repository for MHPI differentiable hydrological models.

A differentiable astrophysics simulator in JAX

Training methodologies for autoregressive neural operators/emulators in JAX.

A library for soft differentiable relaxations of common JAX functions.

Controlling fluids in a reduced-dimensional simulation.

Repository for NeurIPS 2025 paper, "Physics-informed Reduced Order Modeling of Time-dependent PDEs via Differentiable Solvers."

Official code implementation of the paper on adaptive node position in biological transport networks

A high performance data-driven and differentiable SPH fluid solver