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

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

Differentiable Social Force simulation with universal interaction potentials.

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.

A GPU-Parallel analytical physics engine with mujoco API compatibility

differentiable (magneto)hydrodynamics for astrophysics in JAX

TorchFSM: Fourier Spectral Method with PyTorch

Differentiable Ray Tracing Toolbox for Radio Propagation Simulations

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

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

Differentiable Flexible Mechanical Metamaterials

JAX compilation of RDDL description files, and a differentiable planner in JAX.