PyRank

Auto Differentiation Python Packages

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

Unitful Quantities in JAX

17K 69 5
GalacticDynamics
unxt-api

Unitful Quantities in JAX

13K 69 5
GalacticDynamics
unxts-api

Unitful Quantities in JAX

11K 69 5
GalacticDynamics
unxts-parametric

Unitful Quantities in JAX

7K 69 5
GalacticDynamics
unxts-linalg

Unitful Quantities in JAX

7K 69 5
GalacticDynamics
unxts-hypothesis

Unitful Quantities in JAX

6K 69 5
oberbichler
hyperjet

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

3K 17 4
abess-team
skscope

skscope: Sparse-Constrained OPtimization via itErative-solvers

1K 309 16
NVIDIA
minkowskiengine

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

1K 3K 489
adxorg
astrodynx

A modern astrodynamics library powered by JAX: differentiate, vectorize, JIT to GPU/TPU, and more

869 11 6
GalacticDynamics
unxts-interop-gala

Unitful Quantities in JAX

669 69 5
oberbichler
hypergraph

Reversed mode second order automatic differentiation for python (WIP)

656 4 0
GalacticDynamics
galax

Galactic and Gravitational Dynamics in Python (+ GPU and autodiff)

531 51 10
GalacticDynamics
unxts-interop-matplotlib

Unitful Quantities in JAX

323 69 5
GalacticDynamics
unxts-interop-xarray

Unitful Quantities in JAX

317 69 5
GalacticDynamics
unxt-hypothesis

Unitful Quantities in JAX

289 69 5
GalacticDynamics
jax-quantity

Unitful Quantities in JAX

264 69 5
WeiXuanChan
autod

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

155 0 0
mntsx
thoad

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

47 6 1