Efficiently computes derivatives of NumPy code.
☆7,463Mar 10, 2026Updated this week
Alternatives and similar repositories for autograd
Users that are interested in autograd are comparing it to the libraries listed below
Sorting:
- Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more☆35,038Updated this week
- A probabilistic programming language in TensorFlow. Deep generative models, variational inference.☆4,843Mar 18, 2024Updated last year
- A pedagogical implementation of Autograd☆1,013May 26, 2020Updated 5 years ago
- Deep universal probabilistic programming with Python and PyTorch☆8,984Jul 9, 2025Updated 8 months ago
- Flax is a neural network library for JAX that is designed for flexibility.☆7,105Updated this week
- Source-to-Source Debuggable Derivatives in Pure Python☆2,316Sep 29, 2022Updated 3 years ago
- NumPy & SciPy for GPU☆10,824Updated this week
- Autograd automatically differentiates native Torch code☆559Oct 21, 2021Updated 4 years ago
- A Python toolbox for performing gradient-free optimization☆4,157May 30, 2025Updated 9 months ago
- JAX-based neural network library☆3,197Updated this week
- A flexible framework of neural networks for deep learning☆5,922Aug 28, 2023Updated 2 years ago
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,423Feb 20, 2026Updated 3 weeks ago
- A highly efficient implementation of Gaussian Processes in PyTorch☆3,838Mar 3, 2026Updated last week
- NumPy interface with mixed backend execution☆1,101Feb 19, 2018Updated 8 years ago
- TensorFlow-based neural network library☆9,906Feb 10, 2026Updated last month
- Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.☆2,621Feb 28, 2026Updated last week
- A flexible tool for creating, organizing, and sharing visualizations of live, rich data. Supports Torch and Numpy☆10,237Feb 28, 2026Updated last week
- Distributed Asynchronous Hyperparameter Optimization in Python☆7,607Updated this week
- Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.☆14,676Dec 1, 2025Updated 3 months ago
- A scikit-learn compatible neural network library that wraps PyTorch☆6,151Feb 25, 2026Updated 2 weeks ago
- Scalable, Portable and Distributed Gradient Boosting (GBDT, GBRT or GBM) Library, for Python, R, Java, Scala, C++ and more. Runs on sing…☆28,105Updated this week
- Bayesian Modeling and Probabilistic Programming in Python☆9,514Mar 5, 2026Updated last week
- Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.☆6,364Apr 4, 2025Updated 11 months ago
- Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler; for Python, R, Juli…☆20,828Oct 25, 2023Updated 2 years ago
- Intel® Nervana™ reference deep learning framework committed to best performance on all hardware☆3,869Dec 23, 2020Updated 5 years ago
- Deep Learning for humans☆63,977Updated this week
- Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.☆30,906Updated this week
- Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.☆41,617Mar 6, 2026Updated last week
- Bayesian optimization in PyTorch☆3,473Mar 6, 2026Updated last week
- Uniform Manifold Approximation and Projection☆8,113Feb 25, 2026Updated 2 weeks ago
- Spearmint Bayesian optimization codebase☆1,565Dec 27, 2019Updated 6 years ago
- Tensors and Dynamic neural networks in Python with strong GPU acceleration☆98,039Updated this week
- Probabilistic reasoning and statistical analysis in TensorFlow☆4,415Mar 4, 2026Updated last week
- Sacred is a tool to help you configure, organize, log and reproduce experiments developed at IDSIA.☆4,358Oct 22, 2025Updated 4 months ago
- DyNet: The Dynamic Neural Network Toolkit☆3,436Dec 1, 2023Updated 2 years ago
- Gaussian processes in TensorFlow☆1,905May 29, 2025Updated 9 months ago
- High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.☆4,751Updated this week
- Optax is a gradient processing and optimization library for JAX.☆2,204Updated this week
- Lightweight library to build and train neural networks in Theano☆3,865Mar 26, 2022Updated 3 years ago