Differentiable Programming Python Packages

warp-lang

A Python framework for GPU-accelerated simulation, robotics, and machine learning.

jaxopt

Hardware accelerated, batchable and differentiable optimizers in JAX.

taichi

Productive, portable, and performant GPU programming in Python.

pytorch-kinematics

Robot kinematics implemented in pytorch

taichi-nightly

The Taichi Programming Language

torchopt

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

tesseract-core

Universal, autodiff-native software components for Simulation Intelligence. 📦

nextstat

High-performance statistical inference engine — MLE, Bayesian, survival, GLM, time series. Rust core with Python bindings.

optiland

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

memo-lang

A probabilistic programming language for metacognitive modeling

fpt-jax

Fermat path tracing with JAX

s2fft

S2FFT: Differentiable and accelerated spherical transforms

mcfit

multiplicatively convolutional fast integral transforms, implementing FFTLog

neuralogic

PyNeuraLogic lets you use Python to create Differentiable Logic Programs

torchbearer

torchbearer: A model fitting library for PyTorch

neuromancer

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

jaxdf

A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations

jaxphys

Differentiable JAX physics engine for classical, quantum, EM, and statistical mechanics

torch-sla

torch sparse linear algebra library

softjax

A library for soft differentiable relaxations of common JAX functions.

janssen

Differentiable and Modular Optical Microscopy Simulations and Inversions

dsgp4

dSGP4: differentiable SGP4. Supports differentiability, ML integration & embarassingly parallel computations

pytorch-kinematics-ms

Robot kinematics implemented in pytorch

genjax

Probabilistic programming with programmable inference for parallel accelerators.