PyRank

Auto Differentiation Python Packages

Python packages with the GitHub topic auto-differentiation. Sorted by relevance, with stars and monthly downloads.
GalacticDynamics
unxt

Quantities in JAX

7K 60 4
oberbichler
hyperjet

Algorithmic differentiation with hyper-dual numbers in C++ and Python

4K 17 4
NVIDIA
minkowskiengine

Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors

3K 3K 473
GalacticDynamics
unxt-api

Abstract dispatch API definitions for unxt

3K 60 4
GalacticDynamics
unxt-hypothesis

Hypothesis strategies for unxt

2K 60 4
abess-team
skscope

skscope: Sparse-Constrained OPtimization via itErative-solvers

2K 326 16
WeiXuanChan
autod

autoD is a lightweight, flexible automatic differentiation for python3 based on numpy.

2K 0 0
adxorg
astrodynx

A modern astrodynamics library powered by JAX.

1K 10 4
oberbichler
hypergraph

Reversed mode second order automatic differentiation for python (WIP)

1K 4 0
GalacticDynamics
jax-quantity

Unitful Quantities in JAX

335 61 4
GalacticDynamics
galax

Galactic Dynamics in Jax.

269 44 8
mntsx
thoad

Lightweight performat Python 3.12+ automatic differentiation system that leverages PyTorch’s computational graph to compute arbitrary-order partial derivatives.

73 6 1