Top 8 Python automatic-differentiation Projects

pennylane

2 2,113 9.8 Python

PennyLane is a cross-platform Python library for differentiable programming of quantum computers. Train a quantum computer the same way as a neural network.

Project mention: Library for Machine learning and quantum computing | dev.to | 2024-04-27

Pennylane

aesara

1 1,161 7.7 Python

Aesara is a Python library for defining, optimizing, and efficiently evaluating mathematical expressions involving multi-dimensional arrays.
InfluxDB

www.influxdata.com sponsored

Power Real-Time Data Analytics at Scale. Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality.
torchopt

1 496 7.1 Python

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

1 245 4.1 Python

Differentiable Fluid Dynamics Package

Project mention: Show HN: Elodin – A better framework for physics simulation | news.ycombinator.com | 2024-03-06

You are completely correct; right now it is just mechanics that we have built out. But, there isn't any theoretical reason you couldn't use this framework for other types of simulation. In particular, the Monte Carlo runner is super flexible. Since we are based on JAX you can utilize a ton of the tooling that others have built in the physics space like https://github.com/tumaer/JAXFLUIDS or https://github.com/DifferentiableUniverseInitiative/jax_cosm... . The goal right now though is pretty firmly focused on controls engineers and their needs, but we envision this becoming broadly used.

neograd

2 230 0.0 Python

A deep learning framework created from scratch with Python and NumPy
torchquad

1 161 5.8 Python

Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX
tensortrax

1 7 6.0 Python

Math on (Hyper-Dual) Tensors with Trailing Axes
WorkOS

workos.com sponsored

The modern identity platform for B2B SaaS. The APIs are flexible and easy-to-use, supporting authentication, user identity, and complex enterprise features like SSO and SCIM provisioning.
memoized_coduals

2 3 2.6 Python

Shows that it is possible to implement reverse mode autodiff using a variation on the dual numbers called the codual numbers

NOTE: The open source projects on this list are ordered by number of github stars. The number of mentions indicates repo mentiontions in the last 12 Months or since we started tracking (Dec 2020).