A differentiable PDE solving framework for machine learning

large population models

Code repository for our paper DiffCloth: Differentiable Cloth Simulation with Dry Frictional Contact

A JAX-based research framework for differentiable and parallelizable acoustic simulations, on CPU, GPUs and TPUs

Differentiable neuron simulations with biophysical detail on CPU, GPU, or TPU.

Pytorch differentiable molecular dynamics

A fast, differentiable, JIT-free, debugging-friendly finite element library for PyTorch.

[RSS 2021] An End-to-End Differentiable Framework for Contact-Aware Robot Design

Differentiable Social Force simulation with universal interaction potentials.

A GPU-Parallel analytical physics engine with mujoco API compatibility

Code for PPR: Physically Plausible Reconstruction from Monocular Videos (ICCV 23).

Efficient GPU-computing simulation for differentiable crystal plasticity finite element method

A library for soft differentiable relaxations of common PyTorch functions.

Fast and differentiable particle accelerator optics simulation for reinforcement learning and optimisation applications.

Differentiable Ray Tracing Toolbox for Radio Propagation Simulations

differentiable (magneto)hydrodynamics for astrophysics in JAX

TorchFSM: Fourier Spectral Method with PyTorch

A differentiable finite element analysis solver for structural optimization based on JAX

Differentiable Flexible Mechanical Metamaterials

Code repository for our paper SoftMAC: Differentiable Soft Body Simulation with Forecast-based Contact Model and Two-way Coupling with Articulated Rigid Bodies and Clothes