A differentiable PDE solving framework for machine learning

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

large population models

Pytorch differentiable molecular dynamics

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

[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).

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

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

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

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

Differentiable Ray Tracing Toolbox for Radio Propagation Simulations

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

A differentiable block-based time domain system simulation framework.

Simulation and design of shape morphing LCE lattices

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

Demos of Cheetah being used for various applications presented in "Cheetah: Bridging the Gap Between Machine Learning and Particle Accelerator Physics with High-Speed, Differentiable Simulations"

JAX-powered Hi-Fi mocks

Trying out neural cellular automata for different kinds of problems