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

tesseract-core

Universal components for differentiable scientific computing. :package:

torchopt

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

nextstat

Universal differentiable statistical inference engine (Rust + GPU). High-performance Frequentist & Bayesian modeling, RL environments, and end-to-end optimization for Science, Pharma, and Finance

sll-core

SLL-Core: PyTorch library for automatic differentiation through discrete operations via Static Local Linearization.

taichi-nightly

Productive, portable, and performant GPU programming in Python.

memo-lang

A probabilistic programming language for metacognitive modeling

optiland

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

s2fft

S2FFT: Differentiable and accelerated spherical transforms

mcfit

multiplicatively convolutional fast integral transforms, implementing FFTLog

fpt-jax

Fermat path tracing with JAX

torchbearer

A model training library for pytorch

neuralogic

PyNeuraLogic lets you use Python to create Differentiable Logic Programs.

jaxdf

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

tesseract-jax

Execute + differentiate Tesseracts as part of JAX programs, with full support for function transformations like JIT, grad, and more. ⚡

neuromancer

Neural Modules with Adaptive Nonlinear Constraints and Efficient Regularization

torch-sla

torch sparse linear algebra library

differometor

A Differentiable Interferometer Simulator for the Computational Design of Gravitational Wave Detectors

pytorch-kinematics-ms

Robot kinematics implemented in pytorch

foveatorch

Differentiable foveated vision for Deep Learning methods

dsgp4

Differentiable SGP4 prototype