Productive, portable, and performant GPU programming in Python.

A Python framework for accelerated simulation, data generation and spatial computing.

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.

Fast and embedded solvers for nonlinear optimal control and nonlinear model predictive control

Hardware accelerated, batchable and differentiable optimizers in JAX.

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

Swift for TensorFlow Deep Learning Library

Robot kinematics implemented in pytorch

torchbearer: A model fitting library for PyTorch

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

Differentiable Finite Element Method with JAX

Comprehensive optical design, optimization, and analysis in Python, including GPU-accelerated and differentiable ray tracing via PyTorch.

Julia bindings for the Enzyme automatic differentiator

🧩 Shape-Safe Symbolic Differentiation with Algebraic Data Types

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

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

Experimental tensor-typed deep learning

A common interface for quadrature and numerical integration for the SciML scientific machine learning organization

Differentiable Computational Lithogrpahy Framework

S2FFT: Differentiable and accelerated spherical transforms

L1: Tensor Studio — The playground for tensors

A unified end-to-end learning and control framework that is able to learn a (neural) control objective function, dynamics equation, control policy, or/and optimal trajectory in a control system.

Differentiable RayTracing in Julia

Differentiable Programming Algorithms in Modern C++