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.

Compositional Differentiable Programming Library

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

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.

🧩 Shape-Safe Symbolic Differentiation with Algebraic Data Types

Julia bindings for the Enzyme automatic differentiator

Robot kinematics implemented in pytorch

Code for our NeurIPS 2022 paper

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

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

Differentiable Finite Element Method with JAX

PyNeuraLogic lets you use Python to create Differentiable Logic Programs