pmwd

Fast Differentiable Sorting and Ranking

A differentiable physics engine for robotics

State Representation Learning (SRL) zoo with PyTorch - Part of S-RL Toolbox

A spherical convolutional neural network for cosmology (TFv1).

JAX-DIPS is a differentiable interfacial PDE solver.

Differentiable cosmological emulators: the JAX version of CosmoPower

multiplicatively convolutional fast integral transforms, implementing FFTLog

Code for our paper on doing resource allocation with graph neural networks

Code repository for the 2023 MICCAI Paper "Differentiable Beamforming for Ultrasound Autofocusing"

(differentiable) gradient-based optimization on a chemical graph for de novo molecule design/optimization (ICLR 2022)