A pedagogical implementation of Autograd
☆1,013May 26, 2020Updated 5 years ago
Alternatives and similar repositories for autodidact
Users that are interested in autodidact are comparing it to the libraries listed below
Sorting:
- Efficiently computes derivatives of NumPy code.☆7,463Updated this week
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆1,028Dec 17, 2025Updated 2 months ago
- Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more☆35,038Updated this week
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,803Mar 4, 2026Updated last week
- Research language for array processing in the Haskell/ML family☆1,666Jan 5, 2026Updated 2 months ago
- Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.☆2,621Feb 28, 2026Updated last week
- JAX-based neural network library☆3,197Updated this week
- Source-to-Source Debuggable Derivatives in Pure Python☆2,316Sep 29, 2022Updated 3 years ago
- Optax is a gradient processing and optimization library for JAX.☆2,204Updated this week
- JAX - A curated list of resources https://github.com/google/jax☆2,065Jan 20, 2026Updated last month
- ☆927Jan 29, 2026Updated last month
- Flax is a neural network library for JAX that is designed for flexibility.☆7,105Updated this week
- higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…☆1,627Mar 25, 2022Updated 3 years ago
- Fast and Easy Infinite Neural Networks in Python☆2,378Mar 1, 2024Updated 2 years ago
- Extending JAX with custom C++ and CUDA code☆403Aug 18, 2024Updated last year
- Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/☆1,923Feb 23, 2026Updated 2 weeks ago
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,423Feb 20, 2026Updated 3 weeks ago
- Normalizing Flows in Jax☆107Aug 19, 2020Updated 5 years ago
- functorch is JAX-like composable function transforms for PyTorch.☆1,436Aug 21, 2025Updated 6 months ago
- Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.☆6,364Apr 4, 2025Updated 11 months ago
- ☆626Jan 23, 2026Updated last month
- Exponential families for JAX☆76Mar 3, 2026Updated last week
- A highly efficient implementation of Gaussian Processes in PyTorch☆3,838Mar 3, 2026Updated last week
- A generic Monte Carlo method based on the Gumbel-Max trick.☆32May 31, 2016Updated 9 years ago
- ☆774Jan 27, 2024Updated 2 years ago
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆316Mar 5, 2026Updated last week
- Differentiable SDE solvers with GPU support and efficient sensitivity analysis.☆1,708Dec 30, 2024Updated last year
- Understanding computation in artificial and biological recurrent networks through the lens of dynamical systems.☆390May 28, 2021Updated 4 years ago
- AlgoPy is a Research Prototype for Algorithmic Differentation in Python☆88Jul 7, 2024Updated last year
- Code for the paper "Learning Differential Equations that are Easy to Solve"☆287Nov 25, 2021Updated 4 years ago
- ☆153May 25, 2020Updated 5 years ago
- Deep universal probabilistic programming with Python and PyTorch☆8,984Jul 9, 2025Updated 8 months ago
- ☆50Oct 22, 2020Updated 5 years ago
- KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows☆1,160Feb 6, 2026Updated last month
- Probabilistic Torch is library for deep generative models that extends PyTorch☆894May 12, 2024Updated last year
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆302Feb 27, 2026Updated last week
- A Graph Neural Network Library in Jax☆1,466Mar 18, 2024Updated last year
- A probabilistic programming language in TensorFlow. Deep generative models, variational inference.☆4,843Mar 18, 2024Updated last year
- A tiny scalar-valued autograd engine and a neural net library on top of it with PyTorch-like API☆14,944Aug 8, 2024Updated last year