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

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

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

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

[Siggraph Asia 2024] Optimized shock-protecting microstructures