differentiable-programming

Productive, portable, and performant GPU programming in Python.

Swift for TensorFlow

10 differentiable physical simulators built with Taichi differentiable programming (DiffTaichi, ICLR 2020)

A general-purpose probabilistic programming system with programmable inference

High-performance automatic differentiation of LLVM and MLIR.

Pytorch-based framework for solving parametric constrained optimization problems, physics-informed system identification, and parametric model predictive control.

Compositional Differentiable Programming Library

Hardware accelerated, batchable and differentiable optimizers in JAX.

Swift for TensorFlow Deep Learning Library

A high-fidelity 3D face reconstruction library from monocular RGB image(s)

torchbearer: A model fitting library for PyTorch

TorchOpt is an efficient library for differentiable optimization built upon PyTorch.

Robot kinematics implemented in pytorch

🧩 Shape-Safe Symbolic Differentiation with Algebraic Data Types

Julia bindings for the Enzyme automatic differentiator

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

Differentiable Finite Element Method with JAX

Code for our NeurIPS 2022 paper

[ICLR 2021 top 3%] Is Attention Better Than Matrix Decomposition?

PyNeuraLogic lets you use Python to create Differentiable Logic Programs